From 85d0abe6494c7f39ccfad9a5ade5b8eeea2fa025 Mon Sep 17 00:00:00 2001 From: vonavi Date: Mon, 16 Apr 2018 15:03:41 +0300 Subject: [PATCH] Fixes an issue with CUDA v9 from https://github.com/torch/cutorch/issues/797 --- torch/Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/torch/Dockerfile b/torch/Dockerfile index 2b26cea..9dd9f0b 100644 --- a/torch/Dockerfile +++ b/torch/Dockerfile @@ -6,6 +6,9 @@ MAINTAINER Vladimir Ivanov RUN apt-get update && apt-get install -y \ git sudo +# Fixes an issue with CUDA v9 from https://github.com/torch/cutorch/issues/797 +export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" + # Run Torch7 installation scripts RUN git clone https://github.com/torch/distro.git /root/torch --recursive RUN cd /root/torch && bash install-deps && ./install.sh