Comments (4)
I'm pretty sure that it is mostly redundant, but I can't see a way around it since PyTorch requires cudatoolkit
. I have dodged doubling up on CUDNN by not using the CUDNN base image, but I'm pretty sure that having a CUDA base image is required for proper handling of the GPU (i.e. I don't think that a stock Ubuntu base image would work). If you find a way of avoiding the redundancy please let me know.
from docker-pytorch.
It seems like the only way to avoid this is to build from source?
from docker-pytorch.
@jbohnslav Right, that would work but would also require some fairly major changes to this project. I'll probably look into it at some point.
from docker-pytorch.
The NVIDIA CUDA base images are actually very small (e.g. https://hub.docker.com/layers/nvidia/cuda/11.3.1-base-ubuntu20.04/images/sha256-5a8e4366bd66c2734183c8b2e58da108c2e74f9edb6005be7c828786132fef5a), so I'm going to close this issue until someone has a good, concrete suggestion. Building PyTorch from source would complicate things considerably and I'm concerned that there would be no real gain and potentially compatibility issues.
from docker-pytorch.
Related Issues (20)
- this is the issue which it's showing while running the docker file HOT 2
- Cuda not found after pulling docker image at docker pull anibali/pytorch:1.4.0-cuda10.1 HOT 2
- can we have a docker image with torch 1.4+ cuda 10.2 and ubuntu 16.04 HOT 1
- Any ideas regarding reducing Image size? HOT 1
- pytorch 1.3 with cuda 10 on ubuntu 18.04 image HOT 1
- could not install software HOT 5
- python3: can't open file 'script.py': [Errno 2] No such file or directory HOT 4
- CUDA Version Problem HOT 1
- Disable user prompt when building dockerfile HOT 1
- install HOT 1
- docker image failed to build HOT 2
- why do we use miniconda with pytorch cuda image? HOT 2
- Images to work with for Google Cloud SDK HOT 1
- About the permissions of docekr images HOT 1
- Key errors when deriving from 2.0 images & installing packages HOT 2
- How to enter the container with su mode HOT 1
- Size to big: use multi stage docker files HOT 4
- ERROR: failed to solve -> 1.7.0-cuda11.0-ubuntu20.04 HOT 1
- Support arm64 platform image HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from docker-pytorch.