Anaconda+pycharm配置pytorch1.1.0+cuda 9.0+python3.7环境
1.首先按照网上教程安装Anaconda,可参考本人之前的随笔。
安装成功Anaconda之后,打开Anaconda Prompt
2.执行命令:
conda create --name EnvName python=3.7
创建虚拟环境
3.执行命令:
conda info -e
查看已有的虚拟环境
4.执行命令:
activate EnvName
激活/切换到创建的新虚拟环境
5.从清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
下载这torchvision-0.3.0-py37_cu90_1.tar.bz2和pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2两个文件到本地。
6.执行命令:
conda install --offline filePath\torchvision-0.3.0-py37_cu90_1.tar.bz2
conda install --offline filePath\pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2
离线安装这两个包。
7.执行命令:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
安装pytorch,cudatoolkit,之后报错说无法下载torchvision-0.3.0-py37_cu90_1.tar.bz2和pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2
安装成功之后,还需要安装numpy,不然会报ImportError: DLL load failed: 找不到指定的模块。
8.在pycharm中创建新项目,project Interpreter->existing interpreter->…->conda Environment->…->miniconda3\envs\EnvName\python.exe
9.创建项目之后,执行如下代码:
import torch
print(torch.cuda.is_available())
输出true便大功告成
补充:
(1)安装PyTorch出错:CondaVerificationError
原因:缓存了一些过时的包
解决方案:清空缓存的包就好了。
命令如下:conda clean --packages --tarballs
参考:https://www.cnblogs.com/joe-xie/p/12369791.html
(2)Conda - Downloaded bytes did not match Content-Length 问题
解决方案:添加清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
(3)PyTorch的下载命令
conda的:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch #cuda9.0
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch #cuda10.0
conda install pytorch-cpu torchvision-cpu -c pytorch #cpu版本
参考:http://www.mustenaka.cn/index.php/2019/06/12/pytorch_download/