From 51ba2c96bdc3d323f9eadf59e63ee33f7651d13f Mon Sep 17 00:00:00 2001 From: rdanecek Date: Mon, 13 Feb 2023 20:31:39 +0100 Subject: [PATCH] Update README --- README.md | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 5526eb1..b6895c1 100644 --- a/README.md +++ b/README.md @@ -121,8 +121,12 @@ but you can also use plain conda if you want (but it will be slower): conda env create python=3.8 --file conda-environment_py38_cu11_ubuntu.yml ``` +In case the specified pytorch version somehow did not install, try again manually: +```bash +mamba install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch +``` -Note: the environment might contain some packages. If you find an environment is missing then just `conda/mamba`- or `pip`- install it and please notify me. +Note: If you find the environment has a missing then just `conda/mamba`- or `pip`- install it and please notify me. 2) Activate the environment: ```bash