win11配置torch GPU环境（cuda 11.6 or 11.3）_pip install torch==1.11.0+cu113 torchvision==0.12.

作者：Gausst松鼠会 | 2024-03-25 08:51:52

踩

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0

现在很多电脑都已经有显卡的配置，对于学习深度学习的小伙伴，使用一些深度学习框架或者训练模型是必不可少的一个环节，那么本文就带大家从头搭建torch GPU环境

1 确定本机cuda版本

win+R 弹出框中输入cmd，在弹出框里输入nvidia-smi，如下所示，可以看到cuda的版本事11.6

2 下载对应的torch版本

打开PyTorch官网

（1）下载命令：pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

（注：命令用pip还是pip3或者是conda根据你安装的python环境决定）

（2）我个人下载的是11.3的版本，两个都可以，如果是11.3的版本可以用如下链接

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-ur

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