环境设置:
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