环境相关 # 检查环境是否都没有冲突或者污染,按照官网默认安装torch的cuda版本可能与系统的不一致 torch.version.cuda # cuda version 和 nvidia driver version 有依赖关系,尤其是装torch >= 1.13 # https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html cat /proc/driver/nvidia/version cat /usr/local/cuda/version.txt