conda安装之后一般都离不开cuda的使用,本篇内容就详细讲一讲cuda的安装以及环境配置
前言
要确定你的电脑有显卡并且是英伟达显卡
要确定你的电脑有显卡并且是英伟达显卡
要确定你的电脑有显卡并且是英伟达显卡
1.确定cuda版本
在cmd命令窗口输入 nvidia-smi
可以看到,CUDA Version : 12.2,说明我的显卡最高支持cuda12.2只要保证,下载的cuda版本<=12.2就可以了。ps:一般建议更新显卡驱动为最新的驱动,并且选择相近的cuda版本
2.下载cuda
在上面确定了显卡支持的最高cuda版本,结合你的项目选择合适的cuda版本,进入官网CUDA Toolkit Archive | NVIDIA Developer
建议选择exe不选在线安装
下载之后使用默认安装位置即可,若自定义路径,请记住自己的安装位置
3.下载cuDNN
cuDNN是用于深度神经网络的GPU加速库,CUDA实现了可以调用GPU,而cuDNN实现了使CUDA更符合深度神经网络的使用,网上装用CUDA和cuDNN的电脑比只装CUDA的电脑的训练速度快1.5倍。并且一些项目是必须使用cuDNN加速。
进入官网cuDNN Archive | NVIDIA Developer
选择cuda版本支持的cuDNN版本即可,进入下载页面需要注册或者登录nvidia账号才能下载,
下载后将压缩包中的文件解压复制到cuda的安装目录,cuda的默认安装目录为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\版本号,自定义目录的话注意自己的路径
3.设置环境变量
在系统->高级系统设置->环境变量->path 中添加如下路径
添加完成后在cmd命令窗口输入 nvcc -V查看输出结果:
安装中有问题请留言,看到会及时回复