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 进入这个文件所在目录。进入文件所在目录命令
- 进入文件所在盘符如E盘 则cmd输入 E: 回车
- 进入所在目录命令cmd 输入 cd 文件所在位置 如:E:\迅雷下载 回车
- 进行pip 命令安装 pip install torch-2.0.1 cu117-cp311-cp311-win_amd64.whl 回车
- 等待安装完成
三、查验是否安装成功
运行以下代码,返回True 则代表成功
import torch
print(torch.cuda.is_available())
版权声明:本文标题:Torch 安装教程 windows系统 内容由网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.freenas.com.cn/jishu/1733267877h1568808.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论