pytorch安装

注:在训练模型的时候,有时候可能需要不同版本的 torch和torchvision,所以需要配置不同的环境。anconda和pycharm自行安装,接下来在pycharm终端pip安装。

1. torch和torchvision下载

进入pytorch官网,[https://pytorch.org]

Pycharm中切换pytorch的环境和配置的教程详解

进入右下角的网站下载,找到需要的版本,我的版本如下

Pycharm中切换pytorch的环境和配置的教程详解

1.3.0-版本 cp37-python版本3.7 win-Windows系统

Pycharm中切换pytorch的环境和配置的教程详解

2. pycharm终端安装

(1)首先创建一个环境

conda create -n pytorch_1.3 python=3.7

Pycharm中切换pytorch的环境和配置的教程详解

注:有可能出现这个error:You will need to adjust your conda configuration to proceed. Use `conda config --show channels 原因:镜像停止服务了,方法:只要删除所有镜像链接,恢复到默认设置即可 conda config --remove-key channels 输入conda config --show channels 可以查看你的channels

(2)中间会出现一个判断,输入Y ,这样你的环境创建好了,然后安装你的torch和torchvision 通过cd命令进入你下载的环境当中 dir` 可以查看你的该路径下的文件

Pycharm中切换pytorch的环境和配置的教程详解

(3)torch安装

pip install “torch-1.3.0+cpu-cp37-cp37m-win_amd64.whl”`

双引号中的可在dir下复制,因为安装中有一个numpy下载,速度比较慢建议使用清华源下载

pip install -i 清华镜像网址(http) “torch-1.3.0+cpu-cp37-cp37m-win_amd64.whl”

(4)torchvision安装同上

pip install "torchvision-0.4.0+cu92-cp37-cp37m-win_amd64.whl"

注:有可能出现error timed out 原因:网速过慢方法:这个可以解决类似安装一些package可用

pip --default- timeout=100 intall "torchvision-0.4.0+cu92-cp37-cp37m-win_amd64.whl"

Pycharm中切换pytorch的环境和配置的教程详解

(5)终端查看

pip list

Pycharm中切换pytorch的环境和配置的教程详解

可以查看安装的版本,最后在pycharm的解释器下配置你的环境

3.环境切换

复现有些模型可能需要不同的torch和torchvision版本,需要创建新的环境进行切换,终端下输入 conda deactivate 可推出当前的环境,接着可以按照一开始的步骤创建你的环境,在(2)后输入 pip activate pytorch_1.3 这样就进入你创建的环境当中,接着可以安装你需要的版本。

总结

华山资源网 Design By www.eoogi.com
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
华山资源网 Design By www.eoogi.com

RTX 5090要首发 性能要翻倍!三星展示GDDR7显存

三星在GTC上展示了专为下一代游戏GPU设计的GDDR7内存。

首次推出的GDDR7内存模块密度为16GB,每个模块容量为2GB。其速度预设为32 Gbps(PAM3),但也可以降至28 Gbps,以提高产量和初始阶段的整体性能和成本效益。

据三星表示,GDDR7内存的能效将提高20%,同时工作电压仅为1.1V,低于标准的1.2V。通过采用更新的封装材料和优化的电路设计,使得在高速运行时的发热量降低,GDDR7的热阻比GDDR6降低了70%。