前言

即使conda虚拟环境中安装了cuda cuDNN,要使用某些拓展功能,仍然需要系统安装CUDA驱动程序

一、 CUDA

1. 查询CUDA版本

(1) 查看显卡驱动版本。切换到桌面,同时按win和r键,打开运行框,输入nvidia-smi并回车,如下,可以看到我的显卡驱动版本为522.06。

(2) 进入CUDA Toolkit官方文档界面:

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

可以看到,我的显卡驱动版本号为522.06,因此,CUDA12.0以下的版本应该都可以安装,也就是只能安装绿色框里的cuda版本。这里,例如我们安装CUDA 11.8版本的。

2. 下载CUDA

1.在确定了安装的CUDA版本后,比如上文确定的CUDA 11.8,就可以进入CUDA下载界面:

https://developer.nvidia.com/cuda-toolkit-archive

3. 安装CUDA

第一步,双击CUDA安装包。

第二步,直接点击【OK】

第三步,点击【同意并继续】

第四步,选择【精简】并点击【下一步】

安装完成,点击【关闭】即可

4. 检查CUDA是否安装成功

  • 第一步,同时按win和r键,调出运行框,并输入cmd,如下:

  • 第二步,输入nvcc -V,如果是下面这个样子,就说明安装成功了。否则,就说明哪一步肯定出错了,需要重装CUDA

5. 配置CUDA环境变量(nvcc -V失败的情况下)

  • 第一步,在电脑【设置】里搜索环境变量,并点击【编辑系统环境变量】。
  • 第二步,在【系统属性】中的【高级】下点击【环境变量】。
  • 第三步,查看是否存在CUDA的环境变量,如果没有,则添加,添加的内容一般是一下图片位置,如果是自定义目录要修改成自定义目录;如果已经存在环境变量,则不用添加。


2,601 人已阅读

分类于: