linux 中cuda默认安装在/usr/local目录中:
-可以使用命令:
ls -l /usr/local | grep cuda
查看该目录下有哪些cuda版本:
如果输出:
lrwxrwxrwx 1 root root 21 Dec 17 2021 cuda -> /usr/local/cuda-11.3/
drwxr-xr-x 17 root root 4096 Dec 17 2021 cuda-11.3
drwxr-xr-x 3 root root 4096 Dec 17 2021 down_cuda_files
则表示当前机器上安装了11.3 uda 版本,/usr/local/cuda是一个软链接,链接到了/usr/local/cuda-11.3目录,表示当前使用的是cuda-11.3版本。
具体的版本可以通过: nvcc -V
# (V大写) 查看。
import torch
print(torch.__version__) # 查看torch当前版本号
print(torch.version.cuda) # 编译当前版本的torch使用的cuda版本号
print(torch.cuda.is_available()) # 查看当前cuda是否可用于当前版本的Torch,如果输出True,则表示可用
print(torch.backends.cudnn.version())