Преглед на файлове

Update Anaconda installation commands (#205)

Alexander Borzunov преди 2 години
родител
ревизия
37373a66c3
променени са 1 файла, в които са добавени 3 реда и са изтрити 3 реда
  1. 3 3
      README.md

+ 3 - 3
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]
 ```