在虚拟环境中测试 cuda 是否安装成功不能用 nvcc -V。这是因为该命令返回的系统默认的 cuda 版本,而非当前环境下使用的 cuda 版本。我们可以在 python 环境下用 pytorch 命令检查当前环境下执行 python 文件时调用的是哪个 cuda。
import torch
torch.__version__ # 查看 torch 版本
torch.version.cuda # 若 cuda 安装成功则返回版本号,否则失败。
torch.backends.cudnn.version()