Windows端CUDA11.3+CUDNN+pytorch环境搭建

1、显卡驱动的安装

最近,在学习pytorch深度学习,遇到很多的坑,环境配置也出现过问题,忍不住和大家进行分享,现在把环境搭建过程分享给大家。

1.1、查看自己的显卡。

具体操作:我的电脑-——属性——设备管理器——显示适配器

Windows端CUDA11.3+CUDNN+pytorch环境搭建

1.2、驱动的下载、安装。

在找到自己的显卡后,去NVIDIA 官网下载安装驱动

Windows端CUDA11.3+CUDNN+pytorch环境搭建

选择自己的显卡类型,如果你是GTX显卡,有可能只能选择Game Ready,区别很小,能用就行

1.3、显卡驱动安装检测

win+R 输入cmd 打开窗口后输入:nvidia-smi

Windows端CUDA11.3+CUDNN+pytorch环境搭建

如果成功显示显卡数据,恭喜。右上角的11.6,说明我的显卡驱动最高支持11.6的cuda。但我肯定不能安装11.6,因为pytorch最高才支持11.3。

2、CUDA11.3下载安装

CUDA官网

Windows端CUDA11.3+CUDNN+pytorch环境搭建

在这里先等一下,博主当时下载好10.2的CUDA然后打开torch的官网后,发现;

Windows端CUDA11.3+CUDNN+pytorch环境搭建

CUDA10.2现在竟然不能够在Windows上安装了,我只好又重新下了一个CUDA11.3。

Windows端CUDA11.3+CUDNN+pytorch环境搭建

下载完后,开始安装,这是我安装10.2的过程,这个无所谓,默认的话,一路nest就行。

Windows端CUDA11.3+CUDNN+pytorch环境搭建

Windows端CUDA11.3+CUDNN+pytorch环境搭建

也可以自己选安装位置,建议不更改,设置环境变量要用的。

搜索——系统变量—环境变量

Windows端CUDA11.3+CUDNN+pytorch环境搭建

安装完毕会出现两个新的环境变量 CUDA_PATH和CUDA_PATH_V11_3但仍需要添加以下几个环境变量,系统默认安装路径:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.3

添加以下五个变量到环境变量中:

CUDA_SDK_PATH=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.3

CUDA_LIB_PATH=%CUDA_PATH%\lib\x64

CUDA_BIN_PATH=%CUDA_PATH%\bin

CUDA_SDK_BIN_PATH=%CUDA_SDK_PATH%\bin\win64

CUDA_SDK_LIB_PATH=%CUDA_SDK_PATH%\common\lib\x64

之后在,系统变量Path的末尾添加:

%CUDA_LIB_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_BIN_PATH%;

添加以下五个:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\lib\x64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\CUPTI\lib64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.3\bin\win64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.3\common\lib\x64

3、CUDNN安装

CUDNN要与CUDA版本匹配:官网

压缩后的3个文件

Windows端CUDA11.3+CUDNN+pytorch环境搭建

将这三个文件放入 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3,若修改路径了,则找到自己的路径。

CUDA检测

win+R–cmd– nvcc-V

Windows端CUDA11.3+CUDNN+pytorch环境搭建

4、pytorch安装

Windows端CUDA11.3+CUDNN+pytorch环境搭建

检测安装:

依次输入
import torch
torch.cuda.is_available()

Windows端CUDA11.3+CUDNN+pytorch环境搭建

显示True表明下载完成,可以开始你的深度学习之旅了!

Original: https://blog.csdn.net/m0_57757248/article/details/123484569
Author: 喜欢编程的小吴
Title: Windows端CUDA11.3+CUDNN+pytorch环境搭建

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/625786/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球