65 lines
2.6 KiB
Docker
65 lines
2.6 KiB
Docker
# ---------------------------------------------------------------
|
|
# Copyright (c) 2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
#
|
|
# NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual property
|
|
# and proprietary rights in and to this software, related documentation
|
|
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
# distribution of this software and related documentation without an express
|
|
# license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
|
|
# ---------------------------------------------------------------
|
|
|
|
FROM nvcr.io/nvidia/pytorch:21.12-py3
|
|
ARG PYTHON_VERSION=3.8
|
|
ARG WITH_TORCHVISION=1
|
|
RUN python -m pip uninstall torch -y
|
|
RUN python -m pip uninstall torchvision -y
|
|
RUN python -m pip uninstall torchaudio -y
|
|
RUN python -m pip uninstall torchtext -y
|
|
RUN python -m pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
|
|
RUN apt-get update
|
|
RUN apt-get install -y build-essential zsh tmux wget
|
|
RUN apt-get install -y libgl1-mesa-glx libusb-1.0-0-dev
|
|
RUN apt purge --auto-remove cmake
|
|
RUN python -m pip install cmake
|
|
RUN python -m pip install scipy
|
|
#RUN python -m pip install pykeops
|
|
RUN python -m pip install h5py
|
|
RUN python -m pip install wandb
|
|
RUN python -m pip install --ignore-installed PyYAML
|
|
#RUN python -m pip install open3d
|
|
RUN python -m pip install cupy-cuda111
|
|
RUN python -m pip install nestargs
|
|
RUN python -m pip install flatten_dict
|
|
#RUN python -m pip uninstall point_cloud_utils
|
|
#RUN python -m pip install git+https://github.com/fwilliams/point-cloud-utils.git
|
|
RUN conda install -y scikit-image
|
|
RUN apt-get update
|
|
#RUN python -m pip uninstall torch-scatter
|
|
RUN export PATH=/usr/local/cuda-11.1/bin:$PATH
|
|
RUN export CPATH=/usr/local/cuda-11.1/include:$CPATH
|
|
RUN export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
|
|
RUN export DYLD_LIBRARY_PATH=/usr/local/cuda-11.1/lib:$DYLD_LIBRARY_PATH
|
|
|
|
RUN python -m pip install git+https://github.com/openai/CLIP.git
|
|
RUN python -m pip install loguru einops comet_ml calmsize diffusers
|
|
|
|
#COPY environment.yml .
|
|
#COPY activate.sh .
|
|
#
|
|
#RUN apt -y update && apt -y install wget curl vim libgcrypt20 coreutils libgl1-mesa-glx
|
|
#
|
|
#RUN wget https://repo.continuum.io/miniconda/Miniconda3-3.7.0-Linux-x86_64.sh -O ./miniconda.sh
|
|
#
|
|
#RUN chmod ouga+xw ./miniconda.sh
|
|
#RUN bash ./miniconda.sh -b -p ./miniconda
|
|
#
|
|
#ENV PATH /opt/pytorch/miniconda/bin:$PATH
|
|
#
|
|
#RUN conda update conda
|
|
#RUN conda env create -f environment.yml
|
|
## RUN /bin/bash -c "source activate pvcnn_pyt13"
|
|
#RUN conda activate pvcnn_pyt13
|
|
#RUN pip install einops
|
|
#RUN pip install sklearn
|
|
#RUN pip install tensorflow
|