torch 这个包非常大,安装也非常麻烦。
基本可以上,他是深度学习里面重要的一个包,
要安装torch 首先要确定自己的操作系统, 自己的python 版本, 自己的版本, 是要安装CPU版本的 torch, 还是GPU版本的torch
***
我这边是WIN10专业版。Python 3.8.10 。
我自己有泰坦RTX的显卡,所以肯定要装GPU版本的 ,不然浪费硬件资源,
通过指令:cuda --version
我发现我的cuda 版本是:12.2
下面就可以找安装包了(就是那个whl 文件)
https://download.pytorch.org/whl/torch/
当下时间(20240424 1136)找不到cuda12.2 ,最新的cuda121 ,所以先用它, 大概率没问题
开源的东西,版本太多了,一个torch 就有1833个版本 ,干!
这个torch文件 有2G大小 ,在国内的网络环境下, 不实用镜像资源 直接pip 安装 大概率完蛋
所以我这边打算下载whl 文件, 然后本地安装, 即便是如此,因为大量依赖项的原因,估计还是会下载 东西
这个安装太复杂了怪不得网上 远程装 AI环境,要收费100大洋,
下载完毕,大致是这个样子
下面 开始安装
标签:python,torch,whl,版本,GPU,安装 From: https://www.cnblogs.com/wenluderen/p/18154729