本文是在电脑中直接安装cuda10.2的过程,如果是要在anaconda环境中安装的话,可以直接只在虚拟环境中使用命令进行安装,用以下链接中的过程即可。 

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


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

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

Windows——安装cuda10.2和cudnn7.6.5(两者搭配稳定可靠,补丁)

win10安装cuda10.2和对应的cudnn版本(cuda和cuDNN完整安装,环境配置简单)

PyTorch和CUDA版本对应关系

aunauhwindows 10 安装cuda11.0 cuDNN(含有环境变量,环境配置比较复杂,参考本文的即可))


1. anaconda3 中 python版本为 3.8

conda create -n env_name python=3.8

2. pytorch 使用conda命令在anaconda中下载

(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

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搭配稳定可靠

 

 

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

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

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐