1. 安装版本
pytorch版本:2.0.0
torchvision版本:0.15.0
torchaudio版本:2.0.0
cuda版本:cuda11.8
python版本:3.9
# CUDA 11.8
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
Anaconda安装Pytorch2.0.0(GPU版本)步骤_torch 2.0.0 directml-CSDN博客参考的这篇文章 很顺利