Commit c91d7c6d authored by Abraham, Subil's avatar Abraham, Subil
Browse files

adding torch vision

parent d539c8f9

pytorchimage-centos-cuda/Dockerfile

+19 −0
Original line number Diff line number Diff line
@@ -116,6 +116,25 @@ RUN cd ${INSTALL_ROOT} \ 
	&& cd apex \

	&& pip install --no-cache-dir --disable-pip-version-check --global-option="--cpp_ext" --global-option="--cuda_ext" .



# Install Torchvision

RUN cd ${INSTALL_ROOT} \

	&& git clone --depth 1 --branch v0.10.0 https://github.com/pytorch/vision \

	&& cd vision \

	&& dnf install -y libjpeg-turbo-devel \

	&& python setup.py install



## Install DALI (not working)

#RUN cd ${INSTALL_ROOT} \

#	&& git clone --recurse-submodules --shallow-submodules https://github.com/NVIDIA/DALI --depth 1 --branch v1.5.0 \

#	&& cd DALI \ 

#	&& mkdir build \

#	&& cd build \

#	&& conda install -y -c conda-forge opencv libsndfile libtar libavformat \

#	&& cmake -D CMAKE_BUILD_TYPE=Release .. \

#	&& make-j4 \

#	&& make install \

#	&& pip install dali/python



# Install DLPRof



ENTRYPOINT ["/bin/bash"]

pytorchimage-centos-cuda/build.sh

+2 −2
Original line number Diff line number Diff line 
#!/bin/bash

set -e 



env TMPDIR=/projects/tmp podman build -v /projects/tmp:/tmp --ulimit nofile=262144:262144 -f Dockerfile -t code.ornl.gov:4567/76a/olcfbaseimages/pytorchimage-centos-cuda:latest . ;

env TMPDIR=/projects/tmp podman build -v /projects/tmp:/tmp --ulimit nofile=262144:262144 -f Dockerfile -t code.olcf.ornl.gov:4567/nccs-peak/olcfbaseimages/pytorchimage-centos-cuda:latest . ;





podman push code.ornl.gov:4567/76a/olcfbaseimages/pytorchimage-centos-cuda:latest;
 
podman push code.olcf.ornl.gov:4567/nccs-peak/olcfbaseimages/pytorchimage-centos-cuda:latest;