admin 管理员组

文章数量: 887007

一、下载安装包

torch 直接使用pip 下载很慢,我这里建议使用迅雷下载,下载完成后卸载迅雷就可以了
torch官网 因为是外网打开很慢。
打开之后往下翻找到这个位置
说明:
1、下载方式最优先迅雷后面我会附链接。然后使用pip 进行安装。我建议使用pip 安装,网上推荐的各种使用conda 安装需要先下载anaconda 对新手来说这又是一个难点,anaconda过于庞杂。各种环境的设置,够新手喝一壶。所以我直接建议pip 安装,等以后对环境有需求在研究下anaconda。

2、对应的CUDA版本。参照CUDA安装教程 选择对应你CUDA版本的torch,如你电脑安装CUDA12.1 那么这里选择CUDA12.1

3、打开cmd 复制这个pip 下载链接,到cmd回车会自动进行下载安装。前提是已经安装好python,没有安装python的参考这个链接python安装教程

4、可以直接复制这个链接这是torch cu117 到迅雷进行下载,然后再使用pip 命令进行安装这个版本目前适用于CUDA11.1以上的所有版本。适用性较强

二、进行安装

1、使用pip链接下载时等待自动完成安装,即可
2.使用迅雷下载好之后我们获得一个名为torch-2.0.1 cu117-cp311-cp311-win_amd64.whl安装包。
首先需要更新python 命令 python.exe -m pip install --upgrade pip
然后需要换源命令 pip3 config set global.index-url https://pypi.tuna.tsinghua.edu/simple

然后打开cmd 进入这个文件所在目录。进入文件所在目录命令

  1. 进入文件所在盘符如E盘 则cmd输入 E: 回车
  2. 进入所在目录命令cmd 输入 cd 文件所在位置 如:E:\迅雷下载 回车
  3. 进行pip 命令安装 pip install torch-2.0.1 cu117-cp311-cp311-win_amd64.whl 回车
  4. 等待安装完成

三、查验是否安装成功

运行以下代码,返回True 则代表成功
import torch
print(torch.cuda.is_available())

本文标签: 教程 系统 Torch Windows