pytorch-CUDA版环境安装

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表示安装成功!