https://github.com/tf63/docker-pytorch

https://github.com/tf63/docker-pytorch

Last synced: 24 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/tf63/docker-pytorch
• Owner: tf63
• Created: 2023-03-19T05:10:15.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2023-03-19T05:16:42.000Z (over 1 year ago)
• Last Synced: 2024-10-14T03:28:48.141Z (24 days ago)
• Language: Dockerfile
• Size: 4.88 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Docker_PyTorch

- VSCode + PyTorchのDocker環境

- NVIDIA Dockerを使用

## 各種設定

### Python

Linter, Formatterを`.vscode/settings.json`で指定

- requirements.txt: `.devcontainer/requirements.txt`

- Linter: `flake8`

- Formatter: `black`

### CUDA, PyTorchのバージョン

`.devcontainer/Dockerfile`で指定

- CUDA 

    - https://hub.docker.com/r/nvidia/cuda/tags で探す

```

    FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04

```

- PyTorch

    - https://download.pytorch.org/whl/torch_stable.html で探す

```

    RUN pip3 install --no-cache-dir -U pip setuptools wheel \

    && pip3 install --no-cache-dir -r /work/requirements.txt \

    && pip3 install --no-cache-dir torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html 

```

### GRAM

- `.devcontainer/docker-compose.yml`ファイル中の`shm_size`で指定