Pytorch安装时死机（内存溢出）的解决办法

---

一、问题

问题
官方的安装命令如下：
我安装的版本是cuda11.7

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

ps：最新安装命令可以参考官网 https://pytorch.org/get-started/locally/

运行这条命令后1分钟就死机，打开任务管理器发现内存溢出了

---

二、原因

查阅资料后发现应该是pip在首次下载pytorch时进行了缓存，但是由于用户权限原因，再次下载安装时无法从本地缓存执行，也许是某个方法递归执行的原因，最终导致了内存溢出（如有不对希望指正）

---

三、解决办法

在pip install命令中添加 --no-cache-dir，这样就不会使用本地缓存过的pytorch，重新下载文件并安装

pip --no-cache-dir install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

---

四、相关补充

ps：如果项目没有使用虚拟环境命（virtualenv、venv），令结尾还要加上 --user，如下

pip --no-cache-dir install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 --user