https://github.com/ancestor-mithril/pytorch-cuda-dockerfiles
https://github.com/ancestor-mithril/pytorch-cuda-dockerfiles
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/ancestor-mithril/pytorch-cuda-dockerfiles
- Owner: ancestor-mithril
- License: apache-2.0
- Created: 2024-04-18T11:54:37.000Z (about 1 year ago)
- Default Branch: master
- Last Pushed: 2024-04-22T12:35:52.000Z (about 1 year ago)
- Last Synced: 2024-04-22T12:58:28.724Z (about 1 year ago)
- Language: Shell
- Size: 39.1 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# pytorch-cuda-dockerfiles
Adapted from https://github.com/iot-salzburg/gpu-jupyter and specialised the Dockerfile.
## Usage
Build image:
```
git clone https://github.com/ancestor-mithril/pytorch-cuda-dockerfiles.git
cd pytorch-cuda-dockerfiles/build_cuda12.1
docker buildx build --network=host -t cuda12.1:py3.11_torch2.3.0 .
```
Create container from image:
```
docker run --name pytorch_container --gpus all -d -it --ipc=host --network=host --user root -e GRANT_SUDO=yes -v ~/data:/home/jovyan/root/data cuda12.1:py3.11_torch2.3.0
```