From f5b3b82efef84a89d0d30b575d301a9ff22e3979 Mon Sep 17 00:00:00 2001 From: HAN LI Date: Mon, 23 Oct 2023 20:59:48 +0800 Subject: [PATCH] Update install.rst Change torch installation from conda to pip --- docs/source/qs/install.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/qs/install.rst b/docs/source/qs/install.rst index 7e25ce353..71eea657c 100644 --- a/docs/source/qs/install.rst +++ b/docs/source/qs/install.rst @@ -27,12 +27,12 @@ OpenFOAM-7 and ParaView-5.6.0 will be installed in the ``/opt`` directory. conda create -n deepflame python=3.8 conda activate deepflame conda install -c cantera libcantera-devel - conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia + pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 conda install pybind11 conda install -c conda-forge easydict -.. Note:: Please go to PyTorch's official website to check your system compatability and choose the installation command line that is suitable for your platform. +.. Note:: Please go to PyTorch's official website to check your system compatability and choose the installation command line that is suitable for your platform. After installing torch, do check if torch.cuda.is_available() returns true to use GPU for DNN inference! .. Note:: Check your ``Miniconda3/envs/deepflame`` directory and make sure the install was successful (lib/ include/ etc. exist).