win11 中 python11 安装 torch gpu 版本

124 阅读1分钟

因为各种国内源都没有提供新版的 python 和 新版的 torchgpu 版本。所以用国内源不管怎么安装都是 cpu 版本的,也就是使用 torch.cuda.is_available() 返回总是 false

以下是本文的操作环境:

image.png

image.png

image.png

  1. 获取下载地址 pytorch.org/get-started…

image.png

  1. 执行给出的命令
  2. 在执行时会由一个 2.5G 的下载文件会很慢,直接将控制台提示的下载地址复制到迅雷中下载

image.png 4.下载完文件后直接使用 pip 命令安装即可

image.png 5.安装剩余两个依赖

执行命令时候要注意 torch 是否本重新安装,如果被重新安装成了其他版本是不行的。

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