2022最新 pytorch安装方法 GPU版本 python3.9 torch-1.13.0+cu116-cp39 torchvision-0.14.0 亲自安装可用！

青葱年少 • 2023年3月5日下午12:01 • Python • 阅读 575

从官网一键命令下载的pytorch是CPU版本的，我们需要手动安装GPU版本的

如图，这样是不能安装gpu版本的。
在这里插入图片描述

解决方案

这里针对python3.9版本，在此网站https://github.com/pytorch/vision查找读者python版本对应的torch和torchvision。
在这里插入图片描述

然后在这个网址https://download.pytorch.org/whl/torch/里下载，我这里是对应了这两个。千万要对应，不然会报错。
在这里插入图片描述

先进入要安装的python环境，base或者虚拟环境，我这里在虚拟环境里装。（虚拟环境是anaconda里创建的，不知道虚拟环境的搜索下别人的博客先学习一下）。上面下载完的两个文件放在要安装的环境根目录中

  pip install .\torch-1.13.0+cu116-cp39-cp39-win_amd64.whl
  pip install .\torchvision-0.14.0+cu116-cp39-cp39-win_amd64.whl

在这里插入图片描述

如图successfully就可以了。

import torch
print(torch.cuda.is_available())
//True

文章出处登录后可见！

已经登录？立即刷新