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 orchvision-0.3.0-py37_cu90_1.tar.bz2

conda install --offline filePathpytorch-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->…->miniconda3envsEnvNamepython.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/

作者:Hahallo
         
本文版权归作者和博客园共有,未经作者同意不能转载,否则保留追究法律责任的权利。
原文地址:https://www.cnblogs.com/bigyang/p/14351496.html