先打开Anaconda Prompt
新建一个环境,可以取名为pytorch:
conda create --name pytorch python=3.7 anaconda
python版本根据你自己的python版本所设
我首先考虑用的anaconda安装(建议先改源,如清华镜像源),按讲应该去PyTorch下找到对应的安装指令
我cuda9.2,要我去github上找指令?WTF?
网上找的conda安装的指令貌似不行。
只好用pip安装了
pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
注意这里cu92如果你cuda是9.0就是cu90,如果是8.0就是cu80.
但是安装要先下载,下载太慢了啊,经常因为各种原因重新下载。这时你可以浏览器中直接网址访问那个地址,把文件下到本地,然后本地安装,速度很快,我就是这么做的。
例如我是下到本地后移到D:/目录下直接安装。安装成功后尝试输入
import torch
print(torch.__version__)
如果不报错,就是安装成功。
最后需要安装 torchvision
我一开始用的是网上找的指令,
pip3 install torchvision
后来发现不对。
不能找到一个满足需要的版本?
后来发现上面不是说要安装0.4.1的版本吗,这个版本是0.5.0的了。
于是改为如下指令:
pip install torchvision==0.4.1
安装成功!
到这一步还没有完,对于使用iPython、jupyter notebook、jupyter lab的小伙伴,如果就这样打开如jupyter lab,执行import torch还是会报错。
你需要装nb_conda或者ipykernel。我两个都安装了,但ipykernel这条路我只尝试了一半,详见参考文档。
安装nb_conda:
1、进入虚拟环境pytorch
conda activate pytorch
2、在pytorch虚拟环境中安装插件nb_conda
conda install nb_conda
安装 ipykernel:
1、进入虚拟环境
conda activate pytorch
2、安装 IPykernel
conda install ipykernel
3、将虚拟环境加入IPykernel
python -m ipykernel install --user --name=pytorch
安装好之后,要先激活pytorch环境,再打开jupyter lab,否则还是没法import torch。
例如我的jupyter lab要在“D:/人工智能”这个目录下打开,那么每次打开都要在cmd下依次执行以下指令:
cd /d d:\人工智能
activate pytorch
jupyter lab
这样岂不麻烦。
于是可以写一个bat文件,打开记事本,输入以下指令:
call activate pytorch
call jupyter lab
把它保存在“D:/人工智能”这个目录下,重命名后缀名为“.bat”,生成快捷方式放在桌面。这样每次直接在桌面启动就行了。
启动顺利,并且可以选择环境。
大功告成!
参考博客:
pytorch 0.4.1+win10 +python3.66+cuda9.2 安装
Could not find a version that satisfies the requirement torch==1.4.0 (from torchvision)