machine-learning-deep-learning/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-02-20 13:50:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated readme for docker

Browse Source
This commit is contained in:
Aladdin Persson
2023-12-08 15:35:07 +01:00
parent 5ccb93b033
commit c543867340
1 changed files with 32 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
README.md
View File

@@ -21,35 +21,7 @@ In this repository you will find tutorials and projects related to Machine Learn
- [TensorFlow Tutorials](#tensorflow-tutorials)
- [Beginner Tutorials](#beginner-tutorials)
- [Architectures](#CNN-Architectures)
## Docker Setup
Follow the steps below to use Docker for setting up a consistent development environment:
1. **Install Docker**
If you don't have Docker installed, use these links to install Docker for your system:
- [Install Docker Engine](https://docs.docker.com/engine/install/)
2. **Nvidia Container Toolkit (For GPU support)**
If you want to utilize GPU acceleration with CUDA, install the Nvidia Container Toolkit:
- [Nvidia Container Toolkit Installation Guide](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html)
3. **Build the Docker Image**
Navigate to the directory containing the Dockerfile and run:
```bash
docker build -t aladdin-image .
```
4. **Run the Docker Container**
Once the image is built, start the container using the following command:
```bash
docker run -it \
--gpus all \
-v "${PWD}:/code" \
-p 8080:8080 \
--name "aladdin-container" \
--env AUTHENTICATE_VIA_JUPYTER="mytoken" \
aladdin-image
```
- [Docker setup](#docker-setup)
## Machine Learning
@@ -175,3 +147,34 @@ If you have any specific video suggestion please make a comment on YouTube :)
* [GoogLeNet](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/TensorFlow/CNN_architectures/GoogLeNet)
* [ResNet](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/TensorFlow/CNN_architectures/ResNet)
## Docker Setup
Follow the steps below to use Docker for setting up a consistent development environment:
1. **Install Docker**
If you don't have Docker installed, use these links to install Docker for your system:
- [Install Docker Engine](https://docs.docker.com/engine/install/)
2. **Nvidia Container Toolkit (For GPU support)**
If you want to utilize GPU acceleration with CUDA, install the Nvidia Container Toolkit:
- [Nvidia Container Toolkit Installation Guide](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html)
3. **Build the Docker Image**
Navigate to the directory containing the Dockerfile and run:
```bash
docker build -t aladdin-image .
```
4. **Run the Docker Container**
Once the image is built, start the container using the following command:
```bash
docker run -it \
--gpus all \
-v "${PWD}:/code" \
-p 8080:8080 \
--name "aladdin-container" \
--env AUTHENTICATE_VIA_JUPYTER="mytoken" \
aladdin-image
```