Comments (2)
--gpus=all fixed my problem thanks.
from docker-pytorch.
Have you set up nvidia-docker and tested that it works? If it's set up correctly, you should be able to run the following and get proper nvidia-smi output:
$ docker run -it --gpus=all nvidia/cuda:10.1-base-ubuntu16.04 nvidia-smi
Note the inclusion of the --gpus=all
option in the docker run command---you may find that this is required or not depending on how you're set up.
If that doesn't work, then the problem is not specific to this project and you should make sure that you get nvidia-docker working properly first.
from docker-pytorch.
Related Issues (20)
- this is the issue which it's showing while running the docker file 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
- Is CUDA being downloaded twice? HOT 4
- 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.