pytorch 安装

轮子文件下载地址
下载版本

我的版本：CUDA12.0
python 3.7
pip默认源更换为清华镜像
临时更换

pip install package -i https://pypi.tuna.tsinghua.edu.cn/simple

永久使用

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

torch和torchvision对应关系


pytorch卸载
1、使用conda卸载Pytorch

conda uninstall pytorch
conda uninstall libtorch

2、使用pip卸载Pytorch

检验GPU版本是否安装成功

import torch
print(torch.__version__)
print(torch.cuda.is_available())

pytorch安装教程