nvidia --msi
发现驱动版本较低,cuda版本建议的是11.4,或者更低,因为cuda向下兼容
nvcc -V发现
实际版本12.3,由于是服务器,咱也不敢升级驱动,只能摸索着装环境。
试了
pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
不行
又试了
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
成功
如果遇到新建环境时空间不足,或者用conda下载包时,没有空间