(Ubuntu/Window)pytorch验证(pytorch是否安装成功)验证pytorch是否可用(CPU/GPU)
验证pytorch是否可用(CPU/GPU)以下是pytorch是否安装成功的验证代码,看到没有报错正常输出了,那就要恭喜你完成小白的第一步了!分享一下: 在确定你自己显卡驱动后,其实可以通过Pytorch官网的命令直接进行pytorch(GPU/CPU)两个版本的安装,安装失败基本是网络问题!pip install -i “豆瓣源” ---->>>>(可以用这总方式u进行)GPU上测试import torch # 能否调用pyto