10分钟安装好torch的GPU版本（Windows）_torch gpu版本

作者：黑客灵魂 | 2024-08-13 15:27:56

踩

torch gpu版本

1. 确定cuda版本

nvcc -V

在这里插入图片描述

版本为11.8 , 一会下载的版本为cu118

在这里插入图片描述

确定python版本为为3.8，一会下载为cp38

https://download.pytorch.org/whl/torch_stable.html

（我测试了几个版本，发现只要保证cu118， cp38就可以。至于中间的torch版本，torchvision版本。什么版本都可以，因此你只要对应上cuda和python的版本就行）

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

pip install “torch-2.0.0+cu118-cp38-cp38-win_amd64.whl”

在这里插入图片描述

pip install “torchvision-0.15.0+cu118-cp38-cp38-win_amd64.whl”

在这里插入图片描述

python

import torch

print(torch.cuda.is_available())

在这里插入图片描述
显示True,表示安装成功！！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/黑客灵魂/article/detail/975503