win10 + Anaconda 安装 pytorch-gpu【2020版 良心推荐】
win10 + Anaconda 安装 pytorch-gpu
- 一、所需软件
- 1.Anaconda
- 2.cuda 10.2
- 3.cudnn v7.6.5
- 二、安装cuda、cuDNN
- 1.安装cuda：直接安装运行cuda，安装过程时间有点长，要耐心等待哦QAQ
- 2. 安装cuDNN：
- 3. 配置环境变量
- 三、安装pytorch-gpu
- 1.在Anaconda中创建一个pytorch的环境
- 2.激活pytorch环境 ：
- 3.在新建的pytorch环境中安装pytorch-gpu
- 4.测试是否成功安装pytorch
- 四、将pytorch环境添加到jupyter notebook中
- 方法一：简单
- 方法二：可以自由命名
- 1.首先安装ipykernel：
- 2.在虚拟环境下创建kernel文件：
- 3.激活conda环境：
- 4.将环境写入notebook的kernel中：
- 5.打开jupyter notebook
- 五、添加清华镜像源
- 1.添加清华镜像源：
- 2.查看所有镜像源：
- 3.通过清华镜像下载pytorch：
- 六、番外
一、所需软件
首先去pytorch官网查看自己所需要下载cuda、cudnn的版本
- cuda下载官网
- cudnn下载官网（需要注册账号）
这里我给出相应版本百度云的链接，方便大家食用
1.Anaconda
链接：https://pan.baidu.com/s/1ZEFoYo8evR_7eySvvSUwTA
提取码：q6qy
2.cuda 10.2
链接：https://pan.baidu.com/s/15BqEltYMd50mjXhEZQ4pQA
提取码：jlwo
3.cudnn v7.6.5
链接：https://pan.baidu.com/s/1mGnATEDl0DJ06L7gErZ5SA
提取码：0yni
二、安装cuda、cuDNN
【安装前提：具备Visual Studio的环境，不然按照精简模式安装会安装失败】
1.安装cuda：直接安装运行cuda，安装过程时间有点长，要耐心等待哦QAQ
- 测试cuda是否安装成功： 在终端输入：
nvcc -V
2. 安装cuDNN：
- 将cuDNN里面的库文件复制到cuda安装目录相应的文件夹内，即解压压缩包，把压缩包中bin,include,lib中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2目录下对应目录中
- 把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64\cupti64_102.dll
拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
3. 配置环境变量
确保已经将上面4个path添加到系统变量中！
三、安装pytorch-gpu
1.在Anaconda中创建一个pytorch的环境
在Anaconda Prompt 终端输入：conda create -n pytorch
2.激活pytorch环境 ：
在Anaconda Prompt 终端输入：conda activate pytorch
3.在新建的pytorch环境中安装pytorch-gpu
在Anaconda Prompt 终端输入从pytorch官网获得的安装指令：
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
==============================================
**（网不稳定的小伙伴可以通过添加镜像源哦，参见步骤五）**
==============================================
4.测试是否成功安装pytorch
四、将pytorch环境添加到jupyter notebook中
方法一：简单
在Anaconda Prompt 终端依次输入：
activate pytorch
conda install nb_conda_kernels
方法二：可以自由命名
1.首先安装ipykernel：
conda install ipykernel
2.在虚拟环境下创建kernel文件：
conda install -n pytorch ipykernel
3.激活conda环境：
activate pytorch
4.将环境写入notebook的kernel中：
python -m ipykernel install --user --name pytorch --display-name "pytorch"
删除指定kernel：jupyter kernelspec remove pytorch
查看所有kernel：jupyter kernelspec list
5.打开jupyter notebook
五、添加清华镜像源
1.添加清华镜像源：
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/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
删除镜像：conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
2.查看所有镜像源：
conda config --show
3.通过清华镜像下载pytorch：
conda install pytorch torchvision cudatoolkit=10.2
=======================================
至此，大功告成！！！
=======================================
六、番外
在这里向大家推荐两篇不错的文章，比我写的更加清楚
windows10安装pytorch，感觉自己基本踩了所有坑，这里提供最有效的两种方法
Import Error：Jupyter Notebook 中无法运行虚拟环境（pytorch、tensorflow）