一个见习炼丹师的笔记

系统环境相关

查看CUDA版本

cat /usr/local/cuda/version.txt

查看CUDNN版本

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

代码相关

手动设置显卡id

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1' # use GPU 0, 1
文章目录