网上教程很多,但是有的地方不太对
装cudnn时按照官方的来,两行复制命令都应该带*
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
#检查时使用-version的头文件
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJO
检查torch是否能使用GPU时应使用torch.cuda.is_available()
好多都没带.cuda
被坑的一直报错,以为装的不对,重装好几回,浪费大量时间