@smth Why did you stop to support GPUs with CUDA compatibility 3.0? I fixed it by adding 3.0 to this environment variable:

  
      github.com
  
  
    pytorch/pytorch/blob/64e94f02b78df3c90de6944195537a4ae19b4dd4/docker/pytorch/Dockerfile#L28 10

 rm ~/miniconda.sh && \
 /opt/conda/bin/conda install numpy pyyaml scipy ipython mkl mkl-include && \
 /opt/conda/bin/conda install -c soumith magma-cuda90 && \
 /opt/conda/bin/conda clean -ya
ENV PATH /opt/conda/bin:$PATH
# This must be done before pip so that requirements.txt is available
WORKDIR /opt/pytorch
COPY . .


RUN git submodule update --init
RUN TORCH_CUDA_ARCH_LIST="3.5 5.2 6.0 6.1 7.0+PTX" TORCH_NVCC_FLAGS="-Xfatbin -compress-all" \
CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" \
pip install -v .


RUN git clone https://github.com/pytorch/vision.git && cd vision && pip install -v .


WORKDIR /workspace
RUN chmod -R a+w /workspace