服务器与本机上使用anaconda安装CUDA与cudnn方法

服务器与本机上使用anaconda安装CUDA与cudnn方法,第1张

第一 CUDA与cudnn的版本对应

 学习深度学习首先要做的就是安装一堆加快计算的环境,因为深度学习知识与硬件更新快,很多时间段各大框架如:pytorch, tensorflow, paddle等都会新出更好用的深度学习算子,所以导致框架的版本一层层更新迭代,留下不同时间所对应的最低兼容CUDA版本,让小白头疼的环境版本对应~~

点击进入该文章详细记录所有CUDA与cudnn的版本对应,与官方下载资源,要不同的版本可以自己看完去提供的网址下载不同的版本。

第二 看完上文,即可配置正确版本的CUDA与cudnn

 举个例子,我需要训练一个很老的框架,TensorFlow1.1.5。在TensorFlow官方知道最低CUDA为 CUDA: 10.0,  (注: 我本机已经安装CUDA11.4,太高也没能兼容TensorFlow1.1.5,无法使用GPU训练)
 那么就可以用上文学到的版本对应找到CUDA: 10.0所对应的cudnn(作用: 优化加速GPU计算能力)版本为cudnn: 7.6.5


 然后使用anaconda安装上述的两个NVIDIA底层框架了。使用conda指令查看要下载的版本,指令如下:

# 查看cudatoolkit 所有版本
conda search cudatoolkit
# 查看cudnn 所以版本
conda search cudnn


 使用conda指令下载CUDA与cudnn

conda install cudatoolkit==10.1.168 
conda install cudnn==7.6.5

附上下载成功的运行图片:

(注:
  优点:安装简单
  缺点:一般不会带头文件,如果需要做编译,则需要照文章安装安装,版本没上文方法那么全)

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/801382.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-07
下一篇 2022-05-07

发表评论

登录后才能评论

评论列表(0条)

保存