CUDA_VISIBLE_DEVICES=0
查看cuda 版本
cmd输入nvcc --version(Windows10)
cat /usr/local/cuda/version.txt(Linux)
查看cudnn 版本 (Linux)
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
nvidia-smi
刷新并高亮显示状态
watch -n 1 -d nvidia-smi
CUDA_VISIBLE_DEVICES=0
查看cuda 版本
cmd输入nvcc --version(Windows10)
cat /usr/local/cuda/version.txt(Linux)
查看cudnn 版本 (Linux)
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
nvidia-smi
刷新并高亮显示状态
watch -n 1 -d nvidia-smi