python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Pycharm切换pytorch

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

作者:特立独行的小新

这篇文章主要介绍了Pycharm中切换pytorch的环境和配置,本文给大家介绍的非常详细,对大家的工作或学习具有一定的参考借鉴价值,需要的朋友可以参考下

pytorch安装

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

1. torch和torchvision下载

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

pip安装,其他根据就需要直接选择在这里插入图片描述

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

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

在这里插入图片描述

2. pycharm终端安装

(1)首先创建一个环境

conda create -n pytorch_1.3 python=3.7

注:有可能出现这个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` 可以查看你的该路径下的文件

在这里插入图片描述

(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"

在这里插入图片描述

(5)终端查看

pip list

在这里插入图片描述

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

3.环境切换

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

总结

到此这篇关于Pycharm中切换pytorch的环境和配置的文章就介绍到这了,更多相关Pycharm中切换pytorch的环境和配置内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文