windows安装pytorch配置pycharm和Anaconda

环境设置:

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即可

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

如图所示:
第二步
输入

conda create -n python35 python=3.5

解释:建立一个名字为 python35 的环境,对应的python 版本是 3.5
成功激活
查看是否成功
cmd 输入

activate python35
python --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

成功安装如图;
torchvision在线安装
测试一下是否安装好了,GPU版的:先将环境切换到py35
进入 Anaconda Prompt:

activate python35
python
import torch
torch.__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 菜了。
**

  • 方法一:

**

  1. 新建项目
    命名

  2. 找到刚在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(不建议,使用需谨慎
整体配置整个setting

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

每次用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
  • 9
    点赞
  • 72
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值