Ai学习之微调--＞PyTorch环境部署

1.在线安装

安装地址：Previous PyTorch Versions | PyTorch

        根据自己电脑的CUDA版本，选择需要的PyTorch版本，通过命令在线安装

2.离线安装

2.1下载对应安装包

        同样根据自己电脑的CUDA版本，选择需要的PyTorch版本，记录里面的torch、torchvision、torchaudio三个版本，在下面网址下载离线安装包：download.pytorch.org/whl/cu121

网址可根据如下选择进入：每个后缀的网址均可进入

        进入到下载界面，选择如下三个安装包下载

        根据自己需求，选择到对应的版本，分别去下载。（“cu”开头的是GPU版本。“cp” 后面的数字代表 python 版本。“win_amd64”表示适配的操作系统），注意cu不要下载成cpu，仔细检查名称，一 一下载对应的torch2.1.0，torchaudio2.1.0，torchvision0.16.0（torch文件内存很大有2个多G），建议右键点击链接新标签页打开下载。

2.2安装

• 首先打开cmd

• 使用“conda activate xxxxx（之前创建的环境名字）”激活到当前环境

• 再使用cd定位到下载torch的目录中

• 最后依次pip install torch、torchvison、torchaudio（一定要输入完整的名字）

• 全部安装完了试一试输入python进入解释器，再输入import torch。如果没有报错即安装成功。

建议还是使用在线安装