Update PyTorch installation CMD in examples (#435)

tensorchord · Jun 18, 2022 · 274e183 · 274e183
1 parent f8f77f7
commit 274e183
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/conda/build.envd b/examples/conda/build.envd
@@ -18,9 +18,11 @@ custom_channels:
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 """)
  install.conda_packages(name = [
- "pytorch=0.4.1", "cuda90"
+ "pytorch",
+ # "torchvision", # optional
+ "cudatoolkit=11.3",
  ], channel= ["pytorch"])
- base(os="ubuntu20.04", language="python3.5")
+ base(os="ubuntu20.04", language="python3.8")
  install.python_packages(name = [
  "flask"
  ])