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 <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