docker容器中安装pytorch很慢怎么办？（亲测有效）

文章目录

• • docker容器中安装pytorch
  • • 前戏
    • 安装
  • 后记

docker容器中安装pytorch

前戏

官网：https://pytorch.org/get-started/locally/

去官网的目的，是查看我们需要install那几个package

发现我们只需要pip： torch torchvision torchaudio

安装

第一步：交互式进入容器

docker exec -it 容器名 bash

第二步：pip install，使用国内源安装

根据官网下载：很慢

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

使用国内源替换之

pip3 install torch torchvision torchaudio -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

具体详情如下：速度起飞

$ docker exec -it pycharm_test bash # 交互式进入容器

________                               _______________
___  __/__________________________________  ____/__  /________      __
__  /  _  _ \_  __ \_  ___/  __ \_  ___/_  /_   __  /_  __ \_ | /| / /
_  /   /  __/  / / /(__  )/ /_/ /  /   _  __/   _  / / /_/ /_ |/ |/ /
/_/    \___//_/ /_//____/ \____//_/    /_/      /_/  \____/____/|__/


WARNING: You are running this container as root, which can cause new files in
mounted volumes to be created as the root user on your host machine.

To avoid this, run the container by specifying your user's userid:

$ docker run -u $(id -u):$(id -g) args...

root@05464b32beba:/tf# pip3 install torch torchvision torchaudio -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
Looking in indexes: http://pypi.douban.com/simple
Collecting torch
  Downloading http://pypi.doubanio.com/packages/2b/5c/14f3d5bf876d4db062e20baa933dce98ab7dde6599f462099fa2f69cc7e0/torch-1.11.0-cp38-cp38-manylinux1_x86_64.whl (750.6 MB)
     |████████████████████████████████| 750.6 MB 405 kB/s
Collecting torchvision
  Downloading http://pypi.doubanio.com/packages/18/8b/309116c0bdc90ac7036ca331bfec8dd3e6f7f6d00f869fe6552da096e94a/torchvision-0.12.0-cp38-cp38-manylinux1_x86_64.whl (21.0 MB)
     |████████████████████████████████| 21.0 MB 10.5 MB/s
Collecting torchaudio
  Downloading http://pypi.doubanio.com/packages/74/2d/899c31bc274ac49754d3964fce9f4b328bed3a277102e8f872f758f282d2/torchaudio-0.11.0-cp38-cp38-manylinux1_x86_64.whl (2.9 MB)
     |████████████████████████████████| 2.9 MB 2.3 MB/s
Requirement already satisfied: typing-extensions in /usr/local/lib/python3.8/dist-packages (from torch) (4.0.1)
Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.8/dist-packages (from torchvision) (9.0.0)
Requirement already satisfied: requests in /usr/local/lib/python3.8/dist-packages (from torchvision) (2.27.1)
Requirement already satisfied: numpy in /usr/local/lib/python3.8/dist-packages (from torchvision) (1.22.1)
Requirement already satisfied: charset-normalizer~=2.0.0; python_version >= "3" in /usr/local/lib/python3.8/dist-packages (from requests->torchvision) (2.0.11)
Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.8/dist-packages (from requests->torchvision) (2021.10.8)
Requirement already satisfied: urllib3<1.27,>=1.21.1 in /usr/local/lib/python3.8/dist-packages (from requests->torchvision) (1.26.8)
Requirement already satisfied: idna<4,>=2.5; python_version >= "3" in /usr/local/lib/python3.8/dist-packages (from requests->torchvision) (3.3)
Installing collected packages: torch, torchvision, torchaudio
Successfully installed torch-1.11.0 torchaudio-0.11.0 torchvision-0.12.0
WARNING: You are using pip version 20.2.4; however, version 22.0.4 is available.
You should consider upgrading via the '/usr/bin/python3 -m pip install --upgrade pip' command.

后记

尽管本地有md文档，其中有完整的记录，但是还是push到网站上，便于检索。