Pytorch学习笔记【1】：Pytorch在win10下搭建cuda版本的环境

一. 实验环境介绍

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已经安装成功了哦~