cuda与driver版本对应关系
[tensorflow]各个tensorflow版本和CUDA版本对应,以及各个GPU版本CUDA和cuDNN对应
ubuntu18.04 安装多版本cuda,并随时切换使用
用上面的方法安装cudnn后,需要验证是否安装成功。此时如果用,下面的命令没显示任何东西
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
就换这个命令试试
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
安装多版本cuda后可能出现 nvcc -v和nvidia -smi输出cuda版本不一样的情况,这个不必担心,nvidia-smi版本大于nvcc -V的版本不会有什么问题。
跑代码的时候,使用的是CUDA Toolkit,也就是/usr/local/cuda这个文件夹链接到的那个cuda版本,也就是nvcc -v输出的版本。
tensorflow 安装成功后测试
import tensorflow as tf
tf.test.is_gpu_available()
结果
r