Skip to content

Commit e7861de

Browse files
committed
Update readme
1 parent 4a3f510 commit e7861de

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -316,11 +316,11 @@ docker run -p 8080:8080 mltooling/ml-workspace-spark:0.12.1
316316
<details>
317317
<summary>Details (click to expand...)</summary>
318318

319-
> _Currently, the GPU-flavor only supports CUDA 10.1. Support for other CUDA versions might be added in the future._
319+
> _Currently, the GPU-flavor only supports CUDA 11.2. Support for other CUDA versions might be added in the future._
320320
321321
The GPU flavor (`mltooling/ml-workspace-gpu`) is based on our default workspace image and extends it with CUDA 10.1 and GPU-ready versions of various machine learning libraries (e.g., tensorflow, pytorch, cntk, jax). This GPU image has the following additional requirements for the system:
322322

323-
- Nvidia Drivers for the GPUs. Drivers need to be CUDA 10.1 compatible, version `>= 418.39` ([📖 Instructions](https://github.com/NVIDIA/nvidia-docker/wiki/Frequently-Asked-Questions#how-do-i-install-the-nvidia-driver)).
323+
- Nvidia Drivers for the GPUs. Drivers need to be CUDA 11.2 compatible, version `>=460.32.03` ([📖 Instructions](https://github.com/NVIDIA/nvidia-docker/wiki/Frequently-Asked-Questions#how-do-i-install-the-nvidia-driver)).
324324
- (Docker >= 19.03) Nvidia Container Toolkit ([📖 Instructions](https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(Native-GPU-Support))).
325325

326326
```bash
@@ -782,9 +782,9 @@ CMD ["python", "/resources/docker-entrypoint.py", "--code-only"]
782782

783783
The workspace is pre-installed with many popular interpreters, data science libraries, and ubuntu packages:
784784

785-
- **Interpreter:** Python 3.8 (Miniconda 3), Java 11 (OpenJDK), NodeJS 14, Scala, Perl 5
785+
- **Interpreter:** Python 3.8 (Miniconda 3), NodeJS 14, Scala, Perl 5
786786
- **Python libraries:** Tensorflow, Keras, Pytorch, Sklearn, XGBoost, MXNet, Theano, and [many more](https://github.com/ml-tooling/ml-workspace/tree/main/resources/libraries)
787-
- **Package Manager:** `conda`, `pip`, `apt-get`, `npm`, `yarn`, `sdk`, `poetry`, `gdebi`, `mvn` ...
787+
- **Package Manager:** `conda`, `pip`, `apt-get`, `npm`, `yarn`, `sdk`, `poetry`, `gdebi`...
788788

789789
The full list of installed tools can be found within the [Dockerfile](https://github.com/ml-tooling/ml-workspace/blob/main/Dockerfile).
790790

0 commit comments

Comments
 (0)