【debug记录】有gpu，但是 pytorch仍显示 cpu【原来是新电脑没安装cuda】

原来是新电脑没安装cuda，以为安装了pytorch包就可以了。

检查过程：

1. 'nvcc' 不是内部或外部命令，也不是可运行的程序, 说明没有安装cuda。

2. 查看电脑显卡最高支持cuda版本：nvidia-smi
   

3. 安装cuda，选择版本：https://developer.nvidia.com/cuda-toolkit-archive

   查看安装的版本：nvcc -V
   

4. 安装pytorch gpu版本：https://pytorch.org/

5. 查看 pytorch gpu 是否安装成功, 显示True则成功。

import torch
print(torch.__version__)
print(torch.cuda.is_available())