diff --git a/docker/Dockerfile b/docker/Dockerfile
index 831e9e1..3093b06 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -5,6 +5,9 @@ LABEL authors="jaret"
 # Set noninteractive to avoid timezone prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
+# ref https://en.wikipedia.org/wiki/CUDA
+ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0 10.0 12.0"
+
 # Install dependencies
 RUN apt-get update && apt-get install --no-install-recommends -y \
     git \