init
This commit is contained in:
34
.devcontainer/Dockerfile
Normal file
34
.devcontainer/Dockerfile
Normal file
@@ -0,0 +1,34 @@
|
||||
FROM tensorflow/tensorflow:latest-gpu
|
||||
|
||||
# Install zstd, wget, python3-pip, and ngrok dependencies
|
||||
RUN apt-get update && apt-get install -y zstd wget python3-pip unzip
|
||||
|
||||
# Download and set up CUDA repository pin
|
||||
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin && \
|
||||
mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
|
||||
|
||||
# Download and install CUDA repository
|
||||
RUN wget https://developer.download.nvidia.com/compute/cuda/12.3.2/local_installers/cuda-repo-ubuntu2204-12-3-local_12.3.2-545.23.08-1_amd64.deb && \
|
||||
dpkg -i cuda-repo-ubuntu2204-12-3-local_12.3.2-545.23.08-1_amd64.deb && \
|
||||
cp /var/cuda-repo-ubuntu2204-12-3-local/cuda-*-keyring.gpg /usr/share/keyrings/ && \
|
||||
apt-get update && \
|
||||
apt-get -y install cuda-toolkit-12-3
|
||||
|
||||
# Download and install cuDNN repository
|
||||
RUN wget https://developer.download.nvidia.com/compute/cudnn/9.7.1/local_installers/cudnn-local-repo-ubuntu2204-9.7.1_1.0-1_amd64.deb && \
|
||||
dpkg -i cudnn-local-repo-ubuntu2204-9.7.1_1.0-1_amd64.deb && \
|
||||
cp /var/cudnn-local-repo-ubuntu2204-9.7.1/cudnn-*-keyring.gpg /usr/share/keyrings/ && \
|
||||
apt-get update && \
|
||||
apt-get -y install cudnn
|
||||
|
||||
# Set environment variables
|
||||
ENV CUDA_HOME=/usr/local/cuda
|
||||
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
|
||||
|
||||
# Install Python packages
|
||||
RUN pip install sqlalchemy scikit-learn pandas matplotlib kaggle
|
||||
4
.devcontainer/devcontainer.json
Normal file
4
.devcontainer/devcontainer.json
Normal file
@@ -0,0 +1,4 @@
|
||||
{
|
||||
"name": "marketparser",
|
||||
"image": "marketparser:latest"
|
||||
}
|
||||
Reference in New Issue
Block a user