Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
jaydu1 authored Oct 3, 2023
1 parent 301cc92 commit 02fad51
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,14 @@ In case GitHub rendering stops working, [NbViewer](https://nbviewer.jupyter.org/

## Dependency

Our Python package is available on PyPI and the user can install the CPU version with the following command:
Our Python package is available on conda-forge and PyPI and the user can install the CPU version with the following command:

```
>> pip install pyvitae
# using conda with conda-forge channel
>>> conda install -c conda-forge pyvitae
# or using PyPI
>>> pip install pyvitae
```
To enable GPU for TensorFlow, one should install CUDA dependencies and the `tensorflow-gpu` package. We also recommend using `conda`, `miniconda`, or `virtualenv` to manage the Python environment and install the package in a new environment.
After installing all required packages, one can open the Jupyter Notebook via the terminal:
Expand Down

0 comments on commit 02fad51

Please sign in to comment.