注意要先安装好cuda和cudnn，并建立虚拟环境

1.进入conda虚拟环境

2.输入下面的指令

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

或者进入链接下载https://download.pytorch.org/whl/torch_stable.html

（注释：cu118表示cuda11.8，cp39表示python3.9，）

下载完成后，进入conda虚拟环境，切换到下载的文件夹

pip install 文件名

安装完成

本文参考文章http://t.csdnimg.cn/wbAxS