分析原因:显卡算力需要与torch算力匹配
安装较高版本的torch,cuda11以上。目前笔者安装的版本如下:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
分析原因:显卡算力需要与torch算力匹配
安装较高版本的torch,cuda11以上。目前笔者安装的版本如下:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html