From 618ef193783a8f7e4987f0b0511c85793a0fd394 Mon Sep 17 00:00:00 2001 From: Amine Djeghri <32715913+AmineDjeghri@users.noreply.github.com> Date: Thu, 9 May 2024 21:42:11 +0200 Subject: [PATCH] Update cuda_pytorch_install.md --- docs/unix_workflow/cuda_pytorch_install.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/docs/unix_workflow/cuda_pytorch_install.md b/docs/unix_workflow/cuda_pytorch_install.md index 9ba9348..65b6f6d 100644 --- a/docs/unix_workflow/cuda_pytorch_install.md +++ b/docs/unix_workflow/cuda_pytorch_install.md @@ -38,6 +38,9 @@ conda env update -n my-env -f conda-env-gpu.yml; conda activate my-env; ``` #### 2.2 Manual installation +
+ Click here to expand + The next commands need to be run inside WSL or Linux and not Windows : - Install build-essential `sudo apt-get install build-essential` (required by some packages like llama-cpp-python, for example) @@ -72,6 +75,8 @@ cd package_example/requirements; conda env update -n my-env -f conda-env-gpu.yml; conda activate my-env; ``` The `-n my-env` option will supersede the environment name specified within the file. Alternatively, for CPU support only, use the command: `conda env update -n my-env -f conda-env-cpu.yml; conda activate my-env;`. +
+ - Run `nvcc --version; # should be cuda_11.8.r11.8` - Verify the successful installation of PyTorch by executing the following Python code: