Skip to content
Snippets Groups Projects
Commit 99ddc046 authored by Ludovic Noels's avatar Ludovic Noels
Browse files

withh gpu

parent 2fd0c26a
No related branches found
No related tags found
1 merge request!309Master
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment