CUDA下载安装教程-最新文章论坛-语义熔炉网

CUDA下载安装教程

该帖子部分内容已隐藏
付费阅读
已售 1
10
此内容为付费阅读,请付费后查看

第一步:查看是否有支持安装CUDA的NVIDA显卡(命令 control /name Microsoft.DeviceManager)

CUDA下载安装教程插图

第二步:查看是否有NVIDA显卡驱动程序,如果有,就不用安装了(一般刚装完系统都会安装这些驱动),建议自动更新驱动程序一下没有,请下载安装,地址:geforce.cn/drivers,有两种安装方式,自动和手动,选择适合自己电脑的显卡驱动下载,安装很简单,直接下一步就可以,默认系统安装路径。

CUDA下载安装教程插图1

安装完后,可以打开NVIDA控制面板,查看GPU显卡所支持的CUDA版本,具体开始菜单 -》NVIDIA控制面板-》帮助-》系统信息-》组件-》nvidia.dll后面的cuda参数,可以看到,我的显卡支持版本为12.8,所以我下载安装cuda12.8版本。

CUDA下载安装教程插图2

下载与安装CUDA

CUDA下载安装教程插图3

第二步:下载完后,双击

CUDA下载安装教程插图4

第三步:安装cuda

CUDA下载安装教程插图5

CUDA下载安装教程插图6

CUDA下载安装教程插图7

CUDA下载安装教程插图8

CUDA下载安装教程插图9

CUDA下载安装教程插图10

第四步:nvcc –version 命令查看安装是否成功

CUDA下载安装教程插图11

验证deviceQuery和bandwidthTest,在命令窗口运行测试文件,定位到 在cuda安装目录的deviceQuery.exe  bandwidthTest.exe两个地方的Result=PASS则说明通过,反之,Rsult=Fail 则需要重新安装。

CUDA下载安装教程插图12

CUDA下载安装教程插图13

如果以上都没问题,则说明CUDA安装成功,至于Cuda安装成功之后的使用,我们可以在Visual Studio中写C++代码使用,也可以在Anaconda和Pycharm中写Python代码使用,额外下载安装cudnn,CUDNN是一个常见的神经网络层加速库文件,能够很大程度把加载到显卡上的网络层数据进行优化计算,而CUDA就像一个很粗重的加速库,其主要依靠的是显卡。CUDNN需要在有CUDA的基础上进行,CUDNN可以在CUDA基础上加速2倍以上。

二、cuDNN安装

1、下载cuDNN

根据安装好的CUDA版本选择合适的cuDNN版本,注意这里是需要注册登录的:

CUDA下载安装教程插图14

CUDA下载安装教程插图15

2、安装cuDNN

解压文件:

tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

将解压后的头文件和库复制到cuda目录中:

cd cudnn-linux-x86_64-8.9.7.29_cuda11-archive
sudo cp include/cudnn*    /usr/local/cuda/include 
sudo cp lib/libcudnn*    /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*   /usr/local/cuda/lib64/libcudnn*

cuDNN安装完成,查看安装的版本:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

显示:

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 7
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

三、torch-CUDA虚拟环境安装

如果使用pytorch,一般在安装pytorch的同时已经设置好cuda的安装,优先虚拟环境安装的cuda版本:

1、torch+cuda安装

2、cudnn安装

conda install cudnn

3、验证

1)激活虚拟环境

2)python输入“import torch”指令导入torch模块

3)print(torch.version.cuda) 查看cuda版本

4)print(torch.backends.cudnn.version()) 查看cudnn版本

请登录后发表评论

      • 管理员的头像-语义熔炉网管理员徽章-年度人气明星-语义熔炉网等级-LV10-语义熔炉网作者超级版主0