瀏覽代碼

Use Docker bootstrap agent

Support for the library agent may have been removed in the migration
from singularity to apptainer. It may be restored later.
Jim Madge 3 年之前
父節點
當前提交
5bb24228fd
共有 1 個文件被更改,包括 17 次插入17 次删除
  1. 17 17
      base_containers/pytorch/pytorch_cu.def.template

+ 17 - 17
base_containers/pytorch/pytorch_cu.def.template

@@ -1,17 +1,17 @@
-BootStrap: library
-From: ubuntu:20.04
-
-%post
-    apt-get -y update
-
-    # Add universe repository (necessary for python3-pip)
-    apt-get -y install software-properties-common
-    add-apt-repository -y -u universe
-
-    # Install python packages
-    apt-get -y install python3 python3-pip
-
-    apt-get clean
-
-    # Install python dependencies
-    pip3 install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version -f $find_links
+Bootstrap: docker
+From: ubuntu:20.04
+
+%post
+    apt-get -y update
+
+    # Add universe repository (necessary for python3-pip)
+    apt-get -y install software-properties-common
+    add-apt-repository -y -u universe
+
+    # Install python packages
+    apt-get -y install python3 python3-pip
+
+    apt-get clean
+
+    # Install python dependencies
+    pip3 install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version -f $find_links