3359 medium.com/@ changrongko/NV-how-to-check-cuda-and-Cu dnn-version-e 05 aa 21 da f6c
cuda版本
cat /usr/local/cuda/version.txt
cudnn版本
cat/usr/local/cuda/include/Cu dnn.h|grepcudnn _ major-a2
测试系统cuda和cudnn的有效性
test_cuda.py
importtorchprint(2.0 ) device=torch.device (' cuda :0 ' if torch.cuda.is _ available ) ) else 'cpu ' ) assume ch.tensor([2.1] ) xx=x.cuda (打印) xx ) ) cudnntestffit