ubuntu16.04下安装tensorflow-gpu

ubuntu16.04下安装tensorflow-gpu
参考链接https://blog.csdn.net/u014595019/article/details/53732015
这里有几处需要注意:
1.安装CUDA的版本,终端输入 ubuntu-drivers devices 查看本系统对应的cuda推荐版本
参考链接:【ubuntu下查看cuda ,cudnn版本,以及对应tensorflow安装版本_qq_40212975的博客-CSDN博客】https://blog.csdn.net/qq_40212975/article/details/89245685
上述链接也有下载cudnn的链接

2.安装tensorflow的链接:
tensorflow-gpu
tensorflow-cpu

3.注意tensorflow与CUDA的版本对应(很重要,不然会报错):
tensorflow与CUDA版本对应关系
.whl文件下载地址

注意:这里的35/38指的就是python的版本,我这里的CDUA用的10.1,对应tensorflow选的1.14,但是tensorflow1.14不支持python3.8(包含)及以上的版本,所以就用ubuntu16.04自带的python3.5就好,千万不要卸掉。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值