Congying-Wang的博客

文章目录查看命令查看命令Nvidia自带了一个nvidia-smi的命令行工具，会显示GPU使用情况：$ nvidia-smi出现如下图：

文章目录1. 查看CUDA版本2. 查看cudnn版本1. 查看CUDA版本方法1: 查看文件cat /usr/local/cuda/version.txt方法2: 命令nvcc --version2. 查看cudnn版本cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2输出如下：（图中版本...

文章目录1 查看GPU 编号及其使用信息2 方法一：在终端执行程序时指定GPU3 方法二：在Python代码中指定GPU1 查看GPU 编号及其使用信息参考文章Linux下查看NVIDIA的GPU使用情况$ nvidia-smi其中左上侧有0、1、2、3的编号，表示GPU的编号，在后面指定GPU时需要使用这个编号。2 方法一：在终端执行程序时指定GPU$ CUDA_VISIBL...