cuda cudnn GPU

1.版本要匹配***

2. pytorch gpu设置

本文链接：https://blog.csdn.net/weixin_42445501/article/details/81257385

1. 显示当前GPU使用情况 nvidia-smi

2. 指定gpu使用

  2.1 终端指定 CUDA_VISIBLE_DEVICES=2 python xxx.py

  2.2 代码中指定

1. import os

 

1. os.environ["CUDA_VISIBLE_DEVICES"] = "2"

3. 常用命令

1.ubuntu中查看已安装python的版本.

查看python2版本

python2 --version  
 查看python3版本

python3 --version
 查看默认使用python版本

python --version
2.查看cuda版本

cat  /usr/local/cuda/version.txt

3.查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

4.查看tensorflow版本

python

>>>import tensorflow as tf

>>>tf.__version__