From 31713859b5409aef6205f9b01a20dfca230e1611 Mon Sep 17 00:00:00 2001 From: Vadim Peretokin Date: Sun, 6 Aug 2023 09:57:16 +0200 Subject: [PATCH] Update install instructions to cuda 11.8 --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 982a1b5f6..f3a87c6e3 100644 --- a/README.md +++ b/README.md @@ -46,7 +46,7 @@ Petals is a community-run system — we rely on people sharing their GPUs. Y 🐍 **Linux + Anaconda.** Run these commands: ```bash -conda install pytorch pytorch-cuda=11.7 -c pytorch -c nvidia +conda install pytorch pytorch-cuda=11.8 -c pytorch -c nvidia pip install git+https://github.com/bigscience-workshop/petals python -m petals.cli.run_server stabilityai/StableBeluga2 --torch_dtype float16 ``` @@ -115,7 +115,7 @@ Learning more: Here's how to install Petals with [Anaconda](https://www.anaconda.com/products/distribution) on Linux: ```bash -conda install pytorch pytorch-cuda=11.7 -c pytorch -c nvidia +conda install pytorch pytorch-cuda=11.8 -c pytorch -c nvidia pip install git+https://github.com/bigscience-workshop/petals ```