欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Pycharm中切换pytorch的环境和配置

程序员文章站 2022-03-10 09:56:24
...

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
这样就进入你创建的环境当中,接着可以安装你需要的版本。