add section for cuda update
This commit is contained in:
parent
01c8e6c896
commit
5a15a60474
19
README.md
19
README.md
@ -64,7 +64,7 @@ password `asdf`.
|
|||||||
|
|
||||||
With these commands we can see if everything worked well:
|
With these commands we can see if everything worked well:
|
||||||
```bash
|
```bash
|
||||||
docker-compose ps
|
docker ps
|
||||||
docker logs [service-name]
|
docker logs [service-name]
|
||||||
```
|
```
|
||||||
|
|
||||||
@ -171,6 +171,23 @@ Then update the config file as shown below and restart the service.
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Update CUDA to another version
|
||||||
|
|
||||||
|
To update CUDA to another version, change in `Dockerfile.header`
|
||||||
|
the line:
|
||||||
|
|
||||||
|
FROM nvidia/cuda:10.1-base-ubuntu18.04
|
||||||
|
|
||||||
|
and in the `Dockerfile.pytorch` the line:
|
||||||
|
|
||||||
|
cudatoolkit=10.1
|
||||||
|
|
||||||
|
Then re-generate and re-run the image, as closer described above:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
./generate_Dockerfile.sh
|
||||||
|
./start-local.sh -p [port]:8888
|
||||||
|
```
|
||||||
|
|
||||||
## Issues and Contributing
|
## Issues and Contributing
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user