问题一:
在官网使用conda命令安装无法安装GPU版本
找到pip命令安装
# CUDA 11.1
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
安装完成后验证是否安装成功
问题二:
卸载后重新安装,一直卡在Collecting torch,等了很久报错“MemoryError”
解决:在命令中加入 --no-chache-dir
pip install --no-cache-dir torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html