A series of Jupyter notebooks that walk you through the fundamentals of Machine Learning and Deep Learning in Python using Scikit-Learn, Keras and TensorFlow 2.
Apache License 2.0
7.84k
stars
3.14k
forks
source link
Fix Dockerfile.gpu to handle CUDA 11.2.2 correctly #144
This PR addresses an issue with the .gpu Dockerfile where the specified CUDA version caused errors during package installation.
Changes made:
CUDA_BASEargument and set it to11.2.2for the base image.CUDAargument for compatibility with other parts of the Dockerfile.ENV CONDA_DEFAULT_ENVto ensure conda environment activation works as expected.These changes resolve the package installation issues and ensure the Docker image builds successfully.