一. 实验环境介绍
anoconda(python3.7) cuda10.1 pycharm
二. 安装anaconda
到anaconda官网安装即可:
下载好之后点击安装即可。如何检测conda装好了呢?在cmd里面打conda -V,不报错就安好了。
三. 安装cuda
百度搜索cuda download。进入官网,下载。
安装好之后直接打开,可能要等一会 ,不要着急。需要一些准备
准备好之后会开始安装引导
上图的勾勾一定要勾上!!!
下载好之后如何检验呢?
可以通过以下方式:
或者第二种方法:
在这个目录下有nvcc.exe文件
四. 准备工作已经做好,安装pytorch!!!!
配置清华镜像:
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
配置额外镜像:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
安装pytorch:
到pytorch官网,找到如下界面:
注意!!!注意!!!注意!!! 不能完全抄那个命令,因为我们已经配置了清华镜像,命令最后的 -c pytorch不能要,因为-c pytorch就代表使用pytorch官方的镜像源。实际上的命令就是:
conda install pytorch torchvision cudatoolkit=10.1
五. 配置pycharm
安装好pycharm
下面一步很关键!!!记住是找到python.exe这个解释器而不是默认的conda!!!!!
六. 测试
import torch
print(torch.__version__)
print('gpu:',torch.cuda)
说明pytorch已经安装成功了哦~