diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 768d1463..e2ea4b41 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -31,19 +31,8 @@ outlined on that page and do not file a public issue. ### Install Dependencies +Same as in [README](README.md) with the exception of: ``` -conda install pytorch torchvision torchtext cudatoolkit=11.3 -c pytorch-nightly - -# For CPU-only install -conda install pytorch torchvision torchtext cpuonly -c pytorch-nightly -``` - -### Install TorchMultimodal - -``` -git clone --recursive https://github.com/facebookresearch/multimodal.git torchmultimodal -cd torchmultimodal -pip install -r requirements.txt python setup.py develop ``` diff --git a/README.md b/README.md index 13dde6ed..771109b6 100644 --- a/README.md +++ b/README.md @@ -15,16 +15,17 @@ TorchMultimodal requires Python >= 3.8. The library can be installed with or wit 1. Create conda environment ``` - conda create -n torch-multimodal + conda create -n torch-multimodal python= conda activate torch-multimodal ``` 2. Install pytorch, torchvision, and torchtext. See [PyTorch documentation](https://pytorch.org/get-started/locally/). - ``` - conda install pytorch torchvision torchtext cudatoolkit=11.3 -c pytorch-nightly + For now we only support Linux platform. + ``` + conda install pytorch torchvision torchtext cudatoolkit=11.3 -c pytorch-nightly -c nvidia - # For CPU-only install - conda install pytorch torchvision torchtext cpuonly -c pytorch-nightly - ``` + # For CPU-only install + conda install pytorch torchvision torchtext cpuonly -c pytorch-nightly + ``` 3. Download and install TorchMultimodal and remaining requirements. ``` git clone --recursive https://github.com/facebookresearch/multimodal.git torchmultimodal