首先安装torch2.2.2
conda list查看虚拟环境库文件
torch 2.2.2 pypi_0 pypi
对应cuda安装cuda版本pytorch
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch
此处安装nvidia10.2.89,cudnn7.6.5的pytorch1.8.1
首先安装torch2.2.2
conda list查看虚拟环境库文件
torch 2.2.2 pypi_0 pypi
对应cuda安装cuda版本pytorch
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch
此处安装nvidia10.2.89,cudnn7.6.5的pytorch1.8.1