pytorch在linux+ CPU安装过程及pip memoryError问题解决

最新推荐文章于 2025-05-28 08:14:07 发布

原创

最新推荐文章于 2025-05-28 08:14:07 发布 · 1.9k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #人工智能 #python

在Linux上使用pip安装PyTorch时遇到MemoryError，问题根源是缓存导致。通过添加`--no-cache-dir`参数尝试解决，但未成功。最终通过升级pip到最新版解决了安装问题，成功安装PyTorch 1.10.1+cpu，torchvision 0.11.2+cpu和torchaudio 0.10.1+cpu。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装pytorch 官方提供了在各种平台及cpu下安装的命令：

Start Locally | PyTorchd

在linux下cpu安装命令为：

pip3 install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html

需要先安装好python && pip 一般都带的

但是安装过程出现MemoryError的报错提示

# pip3 install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
WARNING: Running pip install with root privileges is generally not a good idea. Try `pip3 install --user` instead.
Collecting torch==1.10.1+cpu
  Downloading https://download.pytorch.org/whl/cpu/torch-1.10.1%2Bcpu-cp36-cp36m-linux_x86_64.whl (199.3MB)
    99% |████████████████████████████████| 199.3MB 12.0MB/s eta 0:00:01Exception:
Traceback (most recent call last):
  File "/usr/lib/python3.6/site-pack