欢迎光临
我们一直在努力

pycharm配置pytorch环境,pycharm怎么用pytorch

环境设置:

windows 10(win其他版本可行)
pycharm 2016.3
python 3.5
Anaconda 3—4.3版(其他版本也行)
cuda 8.0
cudnn5.1
pytorch 0.4.0
查看gpu是否支持cuda

一、Anaconda3+cuda+cudnn安装

具体安装请参照我另一篇博客tensorflow+windows:
安装参照链接,点击此处
注意更改不用安装python3.5的,Anaconda3自带了python3.6,对应的我们只要激活python3.5即可

成功安装Anaconda3打开Anaconda prompt激活python3.5

如图所示:

输入

conda create -n python35 python=3.5

解释:建立一个名字为 python35 的环境,对应的python 版本是 3.5

查看是否成功
cmd 输入

activate python35python –version

二、windows 安装 pytorch #使用conda安装,不支持python2.7 ‘conda,cuda8,python3.5’: conda install pytorch -c pytorch ‘conda,cuda9.0,python3.5’: conda install pytorch cuda90 -c pytorch ‘conda,cuda9.1,python3.5’: conda install pytorch cuda91 ‘conda,cudanone,python3.5’: conda install pytorch-cpu -c pytorch ‘conda,cuda8,python3.6’: conda install pytorch -c pytorch ‘conda,cuda9.0,python3.6’: conda install pytorch cuda90 -c pytorch ‘conda,cuda9.1,python3.6’: conda install pytorch cuda91 -c pytorch ‘conda,cudanone,python3.6′: conda install pytorch-cpu -c pytorch #使用pip安装,不支持python2.7’pip,cudanone,python3.5’: pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl ‘pip,cuda8,python3.5’: pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl ‘pip,cuda9.0,python3.5’: pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl ‘pip,cuda9.1,python3.5’: pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl ‘pip,cudanone,python3.6’: pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl ‘pip,cuda8,python3.6’: pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl ‘pip,cuda9.0,python3.6’: pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl ‘pip,cuda9.1,python3.6’: pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl

安装GPU版本,cmd输入:

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl

解释:cp35表示python3.5版,若使用Anaconda自带python此处改成,具体原因见 pytorch官网

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl

成功安装如图:

安装CPU版本,cmd输入:

pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl

注意:windows下pytorch只能是 ≥ 0.4版

最后需要安装 torchvision

pip install torchvision

成功安装如图;

测试一下是否安装好了,GPU版的:先将环境切换到py35
进入 Anaconda Prompt:

activate python35pythonimport torchtorch.__version__torch.cuda.is_available()a=torch.rand(1,2)b=a.cuda()print(a,b)

成功如图;

三、 配置 pycharm

pycharm 官网安装, 点击此处
本人从vscode 下python 转战pycharm怀抱,主要是pycharm太强大了(vscode下某bug始终解决不了,放pycharm就不是bug),vscode很轻便,占cpu小,有很多便捷小功能,但作为 IDE 菜了。
**

方法一:

**

新建项目

找到刚在Anaconda创建的python35环境
文件夹被隐藏了,找不到Anaconda安装路径

如何找到新建的python35环境(envs),见博客

Anaconda下的python路径,点击此处链接

打开python35,找到 .exe

3. 在 pycharm 工程成功引入torch(pytorch)

测试pycharm引入torch是否成功,见本人另一博客,点击此处

**

方法二(推荐使用,操作简洁):

**

新建项目配置setting:快捷键 Ctrl+Alt+S配置 project Interpreter按照方法一的添加 Add local —python35.exe

**

方法三

**
配置整个pycharm(不建议,使用需谨慎

四、注意事项(新手误区)

每次用pytorch时,先打开Anaconda Prompt,激活python3.5(参照1.3),否则程序会使用Anaconda自带的python3.6,本人不知道两者会不会冲突,考虑到安装的 pytorch 、torchvision 、 pip 等等是在3.5版下面。所以呢 建议每次这样操作一遍,并保持 Prompt 一直处于使用状态,安全起见!!

如果要切回原来的Python环境,执行命令:

deactivate python35

如果要移除Python35环境,可通过执行命令:

conda remove –name python35 –all

赞(0)
【声明】:本博客不参与任何交易,也非中介,仅记录个人感兴趣的主机测评结果和优惠活动,内容均不作直接、间接、法定、约定的保证。访问本博客请务必遵守有关互联网的相关法律、规定与规则。一旦您访问本博客,即表示您已经知晓并接受了此声明通告。