diff --git a/docker/Dockerfile.all-py27 b/docker/Dockerfile.all-py27 index 8c717da..1996481 100644 --- a/docker/Dockerfile.all-py27 +++ b/docker/Dockerfile.all-py27 @@ -65,6 +65,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.all-py27-cpu b/docker/Dockerfile.all-py27-cpu index 848db75..2faff76 100644 --- a/docker/Dockerfile.all-py27-cpu +++ b/docker/Dockerfile.all-py27-cpu @@ -65,6 +65,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.all-py27-jupyter b/docker/Dockerfile.all-py27-jupyter index b718782..4213452 100644 --- a/docker/Dockerfile.all-py27-jupyter +++ b/docker/Dockerfile.all-py27-jupyter @@ -66,6 +66,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.all-py27-jupyter-cpu b/docker/Dockerfile.all-py27-jupyter-cpu index 06f2eb9..5997a1b 100644 --- a/docker/Dockerfile.all-py27-jupyter-cpu +++ b/docker/Dockerfile.all-py27-jupyter-cpu @@ -66,6 +66,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.all-py36 b/docker/Dockerfile.all-py36 index 6ae5104..b0c3c31 100644 --- a/docker/Dockerfile.all-py36 +++ b/docker/Dockerfile.all-py36 @@ -74,6 +74,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.all-py36-cpu b/docker/Dockerfile.all-py36-cpu index 2e67925..a009871 100644 --- a/docker/Dockerfile.all-py36-cpu +++ b/docker/Dockerfile.all-py36-cpu @@ -74,6 +74,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.all-py36-jupyter b/docker/Dockerfile.all-py36-jupyter index c60ee31..120f8fa 100644 --- a/docker/Dockerfile.all-py36-jupyter +++ b/docker/Dockerfile.all-py36-jupyter @@ -75,6 +75,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.all-py36-jupyter-cpu b/docker/Dockerfile.all-py36-jupyter-cpu index f8f4db7..e2971ac 100644 --- a/docker/Dockerfile.all-py36-jupyter-cpu +++ b/docker/Dockerfile.all-py36-jupyter-cpu @@ -75,6 +75,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.torch b/docker/Dockerfile.torch index f2c6024..3687ed5 100644 --- a/docker/Dockerfile.torch +++ b/docker/Dockerfile.torch @@ -32,6 +32,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/docker/Dockerfile.torch-cpu b/docker/Dockerfile.torch-cpu index 920dc7b..d240568 100644 --- a/docker/Dockerfile.torch-cpu +++ b/docker/Dockerfile.torch-cpu @@ -32,6 +32,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # torch # ------------------------------------------------------------------ + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch --recursive && \ cd ~/torch/exe/luajit-rocks && \ diff --git a/generator/modules/torch.py b/generator/modules/torch.py index 2a1aa4c..2961060 100644 --- a/generator/modules/torch.py +++ b/generator/modules/torch.py @@ -9,6 +9,7 @@ class Torch(Module): def build(self): return r''' + export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" $GIT_CLONE https://github.com/torch/distro.git ~/torch''' \ + r''' --recursive && \