Skip to content

Commit

Permalink
Use CUDA 9.0 image in README
Browse files Browse the repository at this point in the history
This is more likely to work for users that did not upgrade their device driver yet.
Also, the image is smaller.
  • Loading branch information
flx42 authored Sep 21, 2018
1 parent 7245090 commit 9f3d92e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

# Test nvidia-smi with the latest official CUDA image
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
```

#### CentOS 7 (docker-ce), RHEL 7.4/7.5 (docker-ce), Amazon Linux 1/2
Expand All @@ -59,7 +59,7 @@ sudo yum install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

# Test nvidia-smi with the latest official CUDA image
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
```
If `yum` reports a conflict on `/etc/docker/daemon.json` with the
`docker` package, you need to use the next section instead.
Expand All @@ -82,7 +82,7 @@ sudo yum install -y nvidia-container-runtime-hook

# Test nvidia-smi with the latest official CUDA image
# You can't use `--runtime=nvidia` with this setup.
docker run --rm nvidia/cuda nvidia-smi
docker run --rm nvidia/cuda:9.0-base nvidia-smi
```

#### Other distributions and architectures
Expand Down

0 comments on commit 9f3d92e

Please sign in to comment.