深度学习环境搭建(3)CUDACUDNN和Tensorflow与Pytorch安装

本文详细介绍了如何在Linux环境下配置CUDA 9.0和10.1,以及安装CUDNN以兼容Tensorflow 1.8和2.2,同时验证Pytorch的兼容性。步骤包括CUDA的选择与安装,CUDNN的配置,以及Tensorflow和Pytorch的一键安装。文章还提供了安装过程中可能遇到的问题及解决方案,如修改Makefile中的CUDA路径和设置环境变量。
摘要由CSDN通过智能技术生成

CUDA环境配置

CUDA版本选择:

关于Linux端的CUDA,目前可以选择的版本有很多,需要注意的是不同版本的Tensorflow对CUDA有版本上的限制,这个限制只有调用GPU版本的Tensorflow才能看出来,或者直接看Tensorflow官网介绍和别人的经验帖也行。
博主推荐电脑安装两个CUDA版本,CUDA9.0和CUDA10.1,这两个版本兼容了Tensorflow1.8和Tensorflow2.2,同时Pytorch也是兼容的(网上说有pytorch对CUDA有限制,实测的时候发现Pytorch对这两个版本都是兼容的)

CUDA安装:

(1)CUDA的下载就不重复了,记得在电脑上预留好对应的存储空间即可,安装文件占用空间比较大一点,两个CUDA加起来有3G以上。
(2)安装命令
Sudo chmod a+X cuda……….sh
Sudo sh cuda…….sh(cuda9.0需要跟上–override)
(3) 安装注意事项

同意用户协议直接Accept

可以选择是否在安装CUDA的同时安装驱动,这个选择要取消掉,或者选No,选上这个会搞砸后面的安装。

CUDA的安装路径选择默认路径即可

关于软链接,如果只安装一个CUDA,可以选择默认值,如果要安装两个CUDA版本,不要使用这个软链接,覆盖安装的时候会冲突,可以不选或者可以选择另一个路径创建软链接。调整一下CUDNN的安装路径即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值