Skip to content

Commit

Permalink
updated dependencies instructions
Browse files Browse the repository at this point in the history
  • Loading branch information
m-tassano authored May 9, 2020
1 parent 5a38f72 commit 1aa6fff
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,17 +25,24 @@ FastDVDnet is orders of magnitude faster than other state-of-the-art methods
The code as is runs in Python +3.6 with the following dependencies:

### Dependencies

You can create a conda environment with all the dependecies by running (Thanks to Antoine Monod for the .yml file)
```
conda env create -f requirements.yml -n <env_name>
```

* [PyTorch v1.0.0](http://pytorch.org/)
* [NVIDIA DALI](https://github.com/NVIDIA/DALI)
*
* [scikit-image](http://scikit-image.org/)
* [numpy](https://www.numpy.org/)
* [OpenCV](https://pypi.org/project/opencv-python/)
* [tensorboardX](https://github.com/lanpa/tensorboardX/)

Note: the tested version of NVIDIA DALI is 0.10.0. To install it (supposing you have CUDA 10.0), you need to run
Note: this project needs the [NVIDIA DALI](https://github.com/NVIDIA/DALI) package to train. The tested version of DALI is 0.10.0. If you prefer to install it yourself (supposing you have CUDA 10.0), you need to run
```
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/10.0 nvidia-dali==0.10.0
```

## Usage

### Testing
Expand Down

0 comments on commit 1aa6fff

Please sign in to comment.