Fixes an issue with CUDA v9 from torch/cutorch#797

vonavi · Apr 16, 2018 · 85d0abe · 85d0abe
1 parent 5a5d19f
commit 85d0abe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch/Dockerfile b/torch/Dockerfile
@@ -6,6 +6,9 @@ MAINTAINER Vladimir Ivanov <ivvl82@gmail.com>
 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