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的安装路径即可。