1.安装CUDA和CUDNN
首先查看电脑是否支持GPU运算。
也就是先看自己电脑显卡的型号:
显卡是SVGA 吧。。
然后去这个地方查看是不是支持GPU 运算的
NAVIDIA官网
所以发现是不行的,看来需要安装CPU版本的pytorch
所以参考了这个:
https://blog.youkuaiyun.com/xinjieyuan/article/details/105100015
和https://blog.youkuaiyun.com/qq_39709535/article/details/81412194
或者可以直接输入一行代码就行吧:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
这两段。
https://blog.youkuaiyun.com/qq_41115379/article/details/107708709
例子测试是否安装成功demo
import torch
print(“hello pytorch{}”.format(torch.version))//返回pytorch版本号
print(torch.cuda.is_available())//返回布尔量是否使用GPU