From 99ddc04645b017d61ffedd060dd974ae14a05ee4 Mon Sep 17 00:00:00 2001 From: noels <l.noels@ulg.ac.be> Date: Fri, 30 Oct 2020 10:05:12 +0100 Subject: [PATCH] withh gpu --- cm3apps/install.txt | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cm3apps/install.txt b/cm3apps/install.txt index 22a84d081..f19eb4404 100644 --- a/cm3apps/install.txt +++ b/cm3apps/install.txt @@ -491,8 +491,14 @@ C) Install Open Cascade (optional) D) Install Torch cd local + // no gpu wget https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.4.0%2Bcpu.zip unzip libtorch-shared-with-deps-1.4.0+cpu.zip + // gpu + sudo apt-get install nvidia-cuda-toolkit + //check version nvcc --version + wget https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.7.0%2Bcu101.zip + add export TORCHDIR=$HOME/local/libtorch in your .bashrc -- GitLab