当前位置: article > 正文

GPU版本pyTorch安装教程_gpu版本的torch和torvision文件

作者：木道寻08 | 2024-08-13 14:16:40

踩

gpu版本的torch和torvision文件

下载GPU版本的torch和torvision
1 GPU版本pytorch和pytorchvision下载地址
2 按照对应关系下，不然用不了，我11.3下的是pytorch 1.8.1和pytorchvision 0.9.1

在这里插入图片描述

3 注意两个地方
cu开头的才是GPU版本的，里面还有CPU开头的，那是CPU版本
cp后面的数字是anconda中python的版本，也要下对应的，我的anconda中python是3.6，所以我下了3.6。注意后面分两个版本，Liunx和Windows

在这里插入图片描述

4 torchvision也跟torch一样，对应着下，网上找对应关系吧，我CUDA11.3，下的0.9.1

在这里插入图片描述

5 下载好放到一个目录，cmd进入那个目录然后pip install 下载好的两个文件，不用一个一个敲，pip install torch，然后按table补齐

在这里插入图片描述

6 torchvision一样的方法，下载完了pip list，看到这两个就成功了

在这里插入图片描述

7 测试

在这里插入图片描述

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