本文是在电脑中直接安装cuda10.2的过程,如果是要在anaconda环境中安装的话,可以直接只在虚拟环境中使用命令进行安装,用以下链接中的过程即可。
全面讲解安装Anaconda+CUDA+PyTorch(含有cuda的具体安装过程)
Windows——安装cuda10.2和cudnn7.6.5(两者搭配稳定可靠,补丁)
win10安装cuda10.2和对应的cudnn版本(cuda和cuDNN完整安装,环境配置简单)
PyTorch和CUDA版本对应关系
aunauhwindows 10 安装cuda11.0 cuDNN(含有环境变量,环境配置比较复杂,参考本文的即可))
conda create -n env_name python=3.8
(1)只安装pytorch,没有安装cuda
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 -c pytorch
(2)先安装了(1),想再次加入cuda的
conda install cudatoolkit=你所需要的版本(例如 cudatoolkit=10.2)
conda install cudnn=你所需要的版本(例如cudnn=7.6.5)
(3)# 有 CUDA 10.2 的, torch和cuda一起安装
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
测试pytorch安装成功的方法:
(1)在当前虚拟环境中输入python
(2)然后输入 import torch
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
在cuDNN下载地址下载,据说cuda10.2和cudnn7.6.5搭配稳定可靠
接下来的安装成功测试根据以下链接中的步骤来。
检测是否安装成功,在文章的最后面