ubuntu20.04+cuda11.1+cudnn8.0安装tensorflow2.4(gpu)

先放参考链接1:

CUDA11.1+cuDNN8.0.4+tensorflow2.4环境配置踩坑记录!_sparkleyn的博客-CSDN博客

链接2:

Ubuntu20.04下深度学习环境配置(持续维护)_m0_37412775的博客-CSDN博客_ubuntu深度学习环境配置

链接1是windows系统,但是可以作为参考,文章中3.3那个错误和链接2中安装tensorflow遇到的那个错误其实是一样的,只不过系统不一样。

pip install tensorflow-gpu==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

这里使用清华源,直接pip install tensorflow会比较慢。链接2里面使用的是豆瓣源(亲测豆瓣确实很快),但是我在使用链接2的指令的时候遇到两个问题,一个是会下载超多版本的tensorflow,然后我修改版本,安装的是tf-nightly2.9,但是安装没报错,我测试import tensorflow一直报error(找不到tensorflow),然后由于比较菜解决不了问题,直接放弃tf--nightly。。。

后面我准备直接下载tensorflow,官网写没有cuda11.1对应的tensorflow。官网:TensorFlow,(tensorflow有中文版官网直接百度就行)。

然后我不想动cuda版本,(因为电脑还有别的pytorch环境,一改就都得改,麻烦)就搜cuda11.1,然后就看到了链接1的类似链接,但是我最开始没用清华源,遇到有一个什么包找不到(我换清华源就可以了,这里搜索说是可能因为网速问题)

测试:

import tensorflow as tf
print(tf.test.is_gpu_available())

返回True,就成功了。

你可能感兴趣的:(pytorch,人工智能,python,深度学习,tensorflow)