Torch CPU
!pip install http://download.pytorch.org/whl/cu75/torch-0.2.0.post1-cp27-cp27mu-manylinux1_x86_64.whl
!pip install torchvision
!pip install --upgrade http://download.pytorch.org/whl/cu75/torch-0.2.0.post1-cp27-cp27mu-manylinux1_x86_64.whl
!pip install --upgrade torchvision
Torch GPU
Build PyTorch from source
RUN git clone GitHub - pytorch/pytorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration
&& cd pytorch
&& git checkout 4eb448a051a1421de1dda9bd2ddfb34396eb7287
&& TORCH_CUDA_ARCH_LIST=“3.5 5.2 6.0 6.1+PTX”
TORCH_NVCC_FLAGS=“-Xfatbin -compress-all”
python setup.py install
&& rm -rf pytorch
Build torch-vision from source
RUN git clone GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision
&& cd vision
&& git checkout 83263d8571c9cdd46f250a7986a5219ed29d19a1
&& python setup.py install
&& rm -rf vision
Refer to my docker for the full script:
https://github.com/QuantScientist/Deep-Learning-Boot-Camp/blob/master/docker/Dockerfile.gpu3