pytorch 环境安装
cpu版本 -不推荐
直接使用pip 安装torch的话,安装的是cpu版本,无法使用cuda,跑起来会很卡
pip3 install torch torchvision torchaudio
cuda 版本
pytorch 官网:https://pytorch.org/
安装cuda版本,需要找到对应的cuda版本的torch
首先确认自己的cuda版本,然后去官网 (https://pytorch.org/) 找到对应的torch版本
如图
如果你的cuda版本是比较新的版本,则一般首页 图片标记1 处就可以直接执行命令下载. 如:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
如果首页没有合适的版本,则从图中标记 2处链接 找到合适自己的版本 https://pytorch.org/get-started/previous-versions/
找到合适自己的版本之后可以用下载工具下载下来. 可以去这里下载 :https://download.pytorch.org/whl/ 或者去其他的下载源下载啊
然后使用本地安装
pip install E:\\pillow-10.2.0-cp312-cp312-win_amd64.whl
判断是否安装成功
# cmd 敲命令进入python
python
import torch
torch.cuda.is_available()
打印true表示安装成功!