try installing pytorch in the conda env. via pip like

当前PyTorch版本是为CUDA 11.3编译的,而系统检测到的CUDA版本是10.1。要解决这个问题,需要安装一个与系统中存在的CUDA版本相匹配的PyTorch版本。

pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html   
 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐