GPU版pytorch的环境安装记录

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）

https://blog.csdn.net/peacefairy/article/details/108015734?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase

如果运行过程出现错误，可以检查数据是否都改为需要的类型。参考如下，tensor(3.1623, device='cuda:0')