pytorch安装遇到的问题

问题一：

在官网使用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