Pycharm中切换pytorch的环境和配置

pytorch安装

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

1. torch和torchvision下载

进入pytorch官网,[https://pytorch.org]
Pycharm中切换pytorch的环境和配置_第1张图片
进入右下角的网站下载,找到需要的版本,我的版本如下
Pycharm中切换pytorch的环境和配置_第2张图片
1.3.0-版本 cp37-python版本3.7 win-Windows系统
Pycharm中切换pytorch的环境和配置_第3张图片

2. pycharm终端安装

(1)首先创建一个环境
conda create -n pytorch_1.3 python=3.7
Pycharm中切换pytorch的环境和配置_第4张图片
注:有可能出现这个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的环境和配置_第5张图片

(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的环境和配置_第6张图片

(5)终端查看

pip list

Pycharm中切换pytorch的环境和配置_第7张图片
可以查看安装的版本,最后在pycharm的解释器下配置你的环境

3.环境切换

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

你可能感兴趣的:(pytorch,深度学习)