anaconda3 下载tf2 cudatookit cundnn

创建并进入虚拟环境

conda create -n name
source activate name

下载tensorflow2-gpu

conda install -c aaronzs tensorflow-gpu
或者
anaconda-navigator 找到自己的虚拟环境然后查找下载（在anaconda图形界面下载会默认下载了cudatookit和cudnn）

下载cudatookit和cudnn

conda install-c anaconda cudatookit
conda install-c anaconda cudnn

检验

import tensorflow as tf
tf.compat.v1.disable_eager_execution()　＃保证sess.run()能够正常运行
a=tf.constant(4.0)
b=tf.constant(6.0)
c=a*b
sess=tf.compat.v1.Session()
print(sess.run©)

测试能否使用GPU：
tf.test.is_gpu_available()

bug

我这里出现driver和runtime版本不一致
因为系统环境下driver是390 而下载是默认最新的cudatookit

所以升级一下系统NVIDIA driver就可以了
ubuntu-drivers devices查看可使用驱动
sudo ubuntu-drivers autoinstall 下载默认推荐驱动，一般是最新版本