Last active
July 7, 2025 15:22
-
-
Save kamwoh/212ece57d61cc7b47b8b745e21160477 to your computer and use it in GitHub Desktop.
My dockerfile for pytorch
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel | |
| RUN apt-get update & apt-get -y install bc unzip git htop vim ffmpeg libsm6 libxext6 libgl1-mesa-glx | |
| RUN pip install jupyterlab | |
| RUN pip install matplotlib | |
| RUN pip install opencv-python | |
| RUN pip install scikit-image | |
| RUN pip install scikit-learn | |
| RUN pip install seaborn | |
| RUN pip install pandas | |
| RUN pip install imutils | |
| RUN pip install tensorflow | |
| RUN pip install kornia | |
| RUN pip install pytorch_memlab | |
| RUN pip install einops | |
| # if use accimage | |
| RUN pip install --prefix=/opt/intel/ipp ipp-devel | |
| RUN pip install git+https://github.com/pytorch/accimage | |
| ENV LD_LIBRARY_PATH=/opt/intel/ipp/lib:$LD_LIBRARY_PATH | |
| # if use pillow-simd | |
| RUN pip uninstall -y pillow | |
| RUN CC="cc -mavx2" pip install -U --force-reinstall pillow-simd | |
| RUN pip install umap-learn | |
| RUN pip install faiss-gpu | |
| RUN pip install wandb | |
| # update torch but staying with cuda11.1 | |
| RUN pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html | |
| RUN pip install timm |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment