原来是新电脑没安装cuda,以为安装了pytorch包就可以了。
检查过程:
'nvcc' 不是内部或外部命令,也不是可运行的程序
, 说明没有安装cuda。查看电脑显卡最高支持cuda版本:
nvidia-smi
安装cuda,选择版本:https://developer.nvidia.com/cuda-toolkit-archive
查看安装的版本:nvcc -V
安装pytorch gpu版本:https://pytorch.org/
查看 pytorch gpu 是否安装成功, 显示
True
则成功。
import torch
print(torch.__version__)
print(torch.cuda.is_available())