在对一些数据进行推理的时候没有必要使用GPU,可以设置其不可见
添加环境变量:
export CUDA_VISIBLE_DEVICES=""
这里设置的是都不可见!
查看驱动版本:
cuda:
cat /usr/local/cuda/version.txt
cudnn:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cuda: 是NVIDIA 退出的通用gpu 计算架构
cuDnn: 是深度学习加速库