1.
安装GPU版本pytorch教程(避坑)
https://blog.csdn.net/weixin_40431584/article/details/105119633
我是直接从后面链接的download的网站下载的torch-1.5.0-cp38-cp38-win_amd64.whl ,再pip install xx。感觉这样更快。
下载安装的1.5.0的torch,直接下载的whl文件。
2.
torch的cuda版本选择是10.2,电脑上cuda版本可以高于torch的版本,我的是cuda11。
win10安装CUDA和cuDNN的正确姿势
https://zhuanlan.zhihu.com/p/94220564?utm_source=wechat_session
3.
cuda版本的代码改写。
PyTorch学习系列(十六)——如何使用cuda进行训练?
https://blog.csdn.net/VictoriaW/article/details/72871777?utm_source=blogxgwz3
GPU 上的 PyTorch - 用 CUDA 训练神经网络(pytorch系列-30)
如果运行过程出现错误,可以检查数据是否都改为需要的类型。参考如下,tensor(3.1623, device='cuda:0')