pytorch下载安装_pytorch镜像源下载

作者：我家自动化 | 2024-04-05 18:19:43

踩

pytorch镜像源下载

1、查看cuda版本

在命令行输入如下代码：

nvidia-smi

本机CUDA版本为12.1

网址为： https://pytorch.org/

选择对应信息，获取到pytorch下载命令

在虚拟环境中输入命令：

注：请输入自己对应的命令

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

下载完成

conda list|findstr torch

进入python环境，导入torch包，如果没有报错，则说明可以正常使用

参考文章：

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