一、离线安装的好处
我们知道,每开发一个AI应用的时候,都需要创建一个conda虚拟环境并且要安装pytorch。这个库如果每新创建一个新的conda环境的时候都需要重新下载一次,那将会消耗非常多的时间。而且国内的环境大家都懂。经常会导致下载失败。所以对于经常开发AI应用的开发者来说,离线安装pytorch是一个非常有用的技能,当然也有其它一些好处
-
网络环境受限时也能安装
- 在无互联网或网络不稳定的环境下使用
- 适合内网开发环境
-
安装速度快
- 避免网络下载延迟
- 一次下载,多次使用
-
版本控制更可靠
- 确保团队使用相同版本
- 避免依赖包版本冲突
二、离线安装步骤
下面以window系统为例子说明安装教程,其它操作系统类似
第一步:查看CUDA版本
先在cmd中执行以下命令行查看CUDA版本
nvidia-smi
执行后如下图所示,我的电脑的CUDA版本是:12.7
第二步:下载torch安装包
- 访问PyTorch官网
- 打开 https://pytorch.org