注意要先安装好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