pytorch、CUDA、cuDNN下载和环境变量设置

2024-05-13 00:37:29

本篇文章的目的是能够在pytorch框架上使用cuda。

电脑本身兼容的CUDA版本

打开NVIDIA控制面板

如下图所示打开系统信息查看

打开组件信息，查看CUDA版本

可以看到本身的CUDA版本是11.6，所以下载CUDA和cuDNN时下载相应版本即可。

pytorch下载

在anaconda下载的pytorch默认是CUP版本的，无法使用GPU，所以为了能够使用GPU，可以输入以下代码，将环境下的pytorch卸载

conda uninstall pytorch

然后去官网下载GPU版本的pytorch

pytorch官网下载

使用pip输入官方给出的代码下载即可。

官网已经不提供CUDA10.2版本pytorch的下载

CUDA的下载

CUDA官方下载

下载完成后双击打开，设置安装目录

然后一路下一步即可。

下载完成后，在cmd中输入

nvcc -V

查看版本

cuDNN下载

cuDNN官网下载

cuDNN的下载需要注册NVIDIA，根据步骤注册即可，如果已经有账户登录即可。

安装与CUDA版本相同的cuDNN

下载完成并解压后cuDNN文件如下所示

将对应文件夹中的文件复制到CUDA文件中

即将上述文件复制到

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

注意一一对应。

上述就完成了相关的设置。

环境变量的设置：

CUDA_BIN_PATH ：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
CUDA_LIB_PATH ：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
CUDA_PATH ：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6
CUDA_PATH_V11_6 ： C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6
CUDA_SDK_PATH：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2

pytorch、CUDA、cuDNN下载和环境变量设置相关推荐

windows 环境下（win10win7）Java8（jdk8）的下载+安装+环境变量设置全过程（超详细图文教程亲测版）
Java8 下载安装环境变量设置全过程一.JDK下载 1.1 官网下载 1.2 下载成果 1.3 下载过程中的注意事项二.JDK安装三.环境变量配置 3.1 JDK在哪儿 3.2 环境变量在哪儿 ...
最新gcc下载和linux环境变量设置
最新gcc下载和linux环境变量设置一.gcc下载提供最新gcc下载路径:点这里跳转 http://ftp.gnu.org/gnu/gcc/ 安装 /gcc_package/configure ...
1.2 JDK安装与环境变量设置
我们会以Java语言来进行脚本编写,所以需要进行JDK安装以及必要的环境变量设置.本书会使用JDK 8的版本,而不会使用JDK 9(太新的有的软件不兼容). 可以直接取Oracle官网选择相应的JDK ...
JDK1.6安装与环境变量设置详细图解
JDK1.6安装图解与环境变量设置为初学者详细介绍了JDK从下载到安装,整个开发伊始的那些事究竟是如何呢?让我们开始吧. 一.JDK1.6安装图解之JDK1.6的下载 JDK现在最新版本是JDK1.6 ...
Linux下环境变量设置
1.在Windows 系统下,很多软件安装都需要配置环境变量,比如安装 jdk ,如果不配置环境变量,在非软件安装的目录下运行javac 命令,将会报告找不到文件,类似的错误. 2.那么什么是环境变 ...
环境变量设置及第三方模块安装
python环境变量设置 '''环境设置1.环境搭建下载安装pythoh2.7.python3+(两者有语法的区别) 配置环境变量:我的电脑=>属性=>高级=>环境变量=>将 ...
indows 平台下 Go 语言的安装和环境变量设置
1. Go 语言 SDK 安装包下载和安装最新稳定版 1.5.3 安装包 go1.5.3.windows-amd64.msi下载地址 https://golang.org/dl/,大小约 69 MB ...
Windows 平台下 Go 语言的安装和环境变量设置
1. Go 语言 SDK 安装包下载和安装最新稳定版 1.5.3 安装包 go1.5.3.windows-amd64.msi下载地址 https://golang.org/dl/,大小约 69 MB ...
JDK、TOMCAT、Ant环境变量设置
JDK.TOMCAT.Ant环境变量设置 JDK环境变量设置第一.下载JDK1.6,现在最新版本是JDK1.6_11,并点击安装. 第二.JDK安装完成后,就进行环境变量的设置. 1.右击" ...
java环境变量设置成功_java环境变量设置
java环境变量设置 java环境变量设置首先,从Sun网站上下载jdk,当前版本为1_5_0_06(其他版本亦可) 最终下载下来的文件为jdk-1_5_0_06-windows-i586-p.ex ...

最新文章

热门文章