From 37373a66c3010c03d259f18e6606ae68bbc22e18 Mon Sep 17 00:00:00 2001 From: Alexander Borzunov Date: Thu, 12 Jan 2023 01:43:25 +0400 Subject: [PATCH] Update Anaconda installation commands (#205) --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 1c5603b..d742123 100644 --- a/README.md +++ b/README.md @@ -38,7 +38,7 @@ for input_ids, labels in data_loader: Run this in an [Anaconda](https://www.anaconda.com) env: ```bash -conda install pytorch cudatoolkit=11.3 -c pytorch +conda install pytorch pytorch-cuda=11.7 -c pytorch -c nvidia pip install -U petals python -m petals.cli.run_server bigscience/bloom-petals ``` @@ -106,7 +106,7 @@ Useful tools and advanced tutorials: Here's how to install Petals with conda: ```bash -conda install pytorch cudatoolkit=11.3 -c pytorch +conda install pytorch pytorch-cuda=11.7 -c pytorch -c nvidia pip install -U petals ``` @@ -122,7 +122,7 @@ __System requirements:__ Petals only supports Linux for now. If you don't have a Petals uses pytest with a few plugins. To install them, run: ```bash -conda install pytorch cudatoolkit=11.3 -c pytorch +conda install pytorch pytorch-cuda=11.7 -c pytorch -c nvidia git clone https://github.com/bigscience-workshop/petals.git && cd petals pip install -e .[dev] ```