From c95f06ae5cf74bf1945c3606a7e51830f1d2bd11 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 15 Jan 2023 18:11:20 +0100 Subject: Fixed Conda env --- environment.yaml | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/environment.yaml b/environment.yaml index 9af40eb..03345c6 100644 --- a/environment.yaml +++ b/environment.yaml @@ -1,15 +1,17 @@ name: ldd channels: - pytorch + - nvidia - xformers/label/dev - defaults dependencies: - cudatoolkit=11.3 + - libcufile=1.4.0.31 - matplotlib=3.6.2 - numpy=1.23.4 - pip=22.3.1 - - python=3.9.15 - - pytorch=1.13.1 + - python=3.10.8 + - pytorch=1.13.1=*cuda* - torchvision=0.14.1 - pip: - -e . -- cgit v1.2.3-70-g09d2