Update Dockerfile

Author: dclong

Dockerfile

@@ -3,5 +3,5 @@ FROM dclong/jupyterhub-cuda
 # GIT: https://github.com/legendu-net/docker-jupyterhub-cuda.git
 
 RUN pip3 install loguru pyarrow pandas \
-    && icon pytorch -ic --cuda 11.1 \
+    && icon pytorch -ic --cuda-version 11.1 \
     && /scripts/sys/purge_cache.sh