Skip to content

Commit

Permalink
add example to download CUDA-enabled LibTorch 2.0.0
Browse files Browse the repository at this point in the history
  • Loading branch information
Massimiliano Bonomi committed Sep 29, 2023
1 parent d63c422 commit 2ee5c95
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion user-doc/Installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -320,7 +320,14 @@ wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-de
unzip libtorch-cxx11-abi-shared-with-deps-2.0.0+cpu.zip ;
\endverbatim

The location of the include and library files need to be exported in the environment:
If you have a GPU, you might want to use the CUDA-accelerated version of LibTorch. For example, the following script downloads the <a href="https://download.pytorch.org/libtorch/cu117/libtorch-shared-with-deps-2.0.0%2Bcu117.zip"> `libtorch-shared-with-deps-2.0.0%2Bcu117.zip`</a> (2.0.0, GPU, Cuda 11.7, pre-cxx11 ABI binary).

\verbatim
wget https://download.pytorch.org/libtorch/cu117/libtorch-shared-with-deps-2.0.0%2Bcu117.zip
unzip libtorch-shared-with-deps-2.0.0+cu117.zip
\endverbatim

In both CPU and GPU cases, the location of the include and library files need to be exported in the environment:

\verbatim
LIBTORCH=${PWD}/libtorch
Expand Down

1 comment on commit 2ee5c95

@PlumedBot
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Found broken examples in automatic/a-masterclass-22-09.txt
Found broken examples in automatic/a-masterclass-22-11.txt
Found broken examples in automatic/a-masterclass-22-12.txt
Found broken examples in automatic/performance-optimization.txt
Found broken examples in automatic/a-trieste-6.txt
Found broken examples in automatic/munster.txt
Found broken examples in automatic/ANN.tmp
Found broken examples in automatic/EDS.tmp
Found broken examples in automatic/EMMI.tmp
Found broken examples in automatic/ENVIRONMENTSIMILARITY.tmp
Found broken examples in automatic/FOURIER_TRANSFORM.tmp
Found broken examples in automatic/FUNCPATHGENERAL.tmp
Found broken examples in automatic/FUNCPATHMSD.tmp
Found broken examples in automatic/FUNNEL.tmp
Found broken examples in automatic/FUNNEL_PS.tmp
Found broken examples in automatic/GHBFIX.tmp
Found broken examples in automatic/INCLUDE.tmp
Found broken examples in automatic/MAZE_MEMETIC_SAMPLING.tmp
Found broken examples in automatic/MAZE_OPTIMIZER_BIAS.tmp
Found broken examples in automatic/MAZE_RANDOM_ACCELERATION_MD.tmp
Found broken examples in automatic/MAZE_RANDOM_WALK.tmp
Found broken examples in automatic/MAZE_SIMULATED_ANNEALING.tmp
Found broken examples in automatic/MAZE_STEERED_MD.tmp
Found broken examples in automatic/PIV.tmp
Found broken examples in automatic/PLUMED.tmp
Found broken examples in MiscelaneousPP.md

Please sign in to comment.