Update Dockerfile
dockerfile updated
This commit is contained in:
@@ -27,8 +27,8 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
|
||||
ENV PATH=${CUDA_HOME}/bin:${PATH}
|
||||
|
||||
# Verify the installation
|
||||
RUN nvcc --version
|
||||
RUN nvidia-smi
|
||||
# RUN nvcc --version
|
||||
# RUN nvidia-smi
|
||||
|
||||
# Install Python packages
|
||||
RUN pip install sqlalchemy scikit-learn pandas matplotlib kaggle
|
||||
|
||||
Reference in New Issue
Block a user