记录贴
由于清华anaconda源无法使用,所以在文章结尾补充离线安装pytorch的方法。
原本已经在anaconda(python3.6)下安装好了pytorch,现在需要python3.7的环境。
cuda是之前安装好的,这里不介绍。
一、创建环境
打开Anaconda Prompt,创建名为py37的环境。
conda create --name py37 python=3.7
二、安装pytorch
注意替换清华源,否则直接安装速度会很慢。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
使用activate py37命令进入py37环境,然后安装pytorch1.0.1。注意不要使用官网给出的命令。
cuda版本根据自己情况修改,我的是9.0。
conda install pytorch torchvision cudatoolkit=9.0
安装结束后,可以使用以下命令进行测试,不报错即安装成功:
离线安装方法:
在此链接中,博主提供了百度云盘的pytorch包。也可去官网下载对应版本的包。
进入Anaconda Prompt,切换到安装包所在目录下,然后使用如下命令安装。
conda install -n name --use-local pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2
其中,-n name的意思是将pytorch安装到虚拟环境“name”中,要根据自己的情况修改。
安装结束后,再继续安装torchvision。
进入pytorch所在的虚拟环境,然后使用命令pip install torchvision即可。
测试: