python3.8+pytorch 1.7.0+cuda10.2+cuDNN

在anaconda虚拟环境中安装pytorch和cuda教程

安装过程中看到的不错的教程

全面讲解安装Anaconda+CUDA+PyTorch（含有cuda的具体安装过程）

Windows——安装cuda10.2和cudnn7.6.5（两者搭配稳定可靠，补丁）

win10安装cuda10.2和对应的cudnn版本(cuda和cuDNN完整安装）

PyTorch和CUDA版本对应关系

windows 10 安装cuda11.0 cuDNN（含有环境变量）

1. anaconda3 中 python版本为 3.8 2. pytorch 使用conda在anaconda中下载

（1）只安装pytorch，没有安装cuda

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cpuonly -c pytorch

（2）先安装了（1），想再次加入cuda的

conda install cudatoolkit=你所需要的版本（例如 cudatoolkit=11.0）

conda install cudnn=你所需要的版本（例如cudnn=7.6.5）

（3）# 有 CUDA 11.0 的， torch和cuda一起安装
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch

测试pytorch安装成功的方法：

（1）在当前虚拟环境中输入python

（2）然后输入 import torch

3. cuda 10.2

pytorch1.7.0版本的支持cuda10.2，所以本次下载的是cuda10.2

从 cuda下载地址选择cuda10.2进行下载

下载的很慢

先安装主要文件，cuda_10.2.89_441.22_win10

安装的时候要关掉360，避免安装失败。

选择自定义安装，只选第一个选项，其他默认即可。

安装完成后，在命令行窗口（cmd）中，输入 nvcc -V 命令进行测试。

 添加环境变量

右键单击此电脑——属性——高级系统设置——环境变量——系统变量

添加两个路径，以下是安装的路径（默认的）：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64

4. cuDNN

在cuDNN下载地址下载，据说cuda10.2和cudnn7.6.5搭配稳定可靠

 

 

 接下来的安装成功测试根据以下连接中的步骤来。

 检测是否安装成功，在文章的最后面