查看对应版本
官网查看
下载包
tensorflow
下载代码
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.6.0
安装cuda
安装cudnn
把对应目录解压到cuda文件夹下即可,不要按照上面的网址安装!
测试代码
返回True即可,警告可忽略。
import tensorflow as tf
tf.test.is_gpu_available()
官网查看
tensorflow
下载代码
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.6.0
把对应目录解压到cuda文件夹下即可,不要按照上面的网址安装!
返回True即可,警告可忽略。
import tensorflow as tf
tf.test.is_gpu_available()