From 1a8b034f18219a7a783453478469f0442c1dfdc5 Mon Sep 17 00:00:00 2001 From: rhprasad0 Date: Wed, 31 Jul 2024 11:36:42 -0400 Subject: [PATCH] Have conda pull pytorch-cuda=11.8 --- .gitignore | 2 +- README.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index a458ea0..daf65c7 100755 --- a/.gitignore +++ b/.gitignore @@ -1,6 +1,6 @@ checkpoints/ __pycache__/ -patterncom/ +PatterNet/ results/ models/ metrics.csv diff --git a/README.md b/README.md index 731efd9..2f14c7c 100644 --- a/README.md +++ b/README.md @@ -68,7 +68,7 @@ Create this environment for our experiments: ```bash conda create -n rscir python=3.9 -y conda activate rscir -conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia +conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia pip install open_clip_torch ```