Commit 991cee2c authored by nd-02110114's avatar nd-02110114
Browse files

⬆️ update Torch versions

parent 086d7d29
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -13,7 +13,7 @@ DeepChem currently supports Python 3.5 through 3.7 and requires these packages o
- `SciPy`_
- `TensorFlow`_

  - `deepchem>=2.4.0` requires tensorflow v2 (2.3.0)
  - `deepchem>=2.4.0` requires tensorflow v2 (2.2.0)
  - `deepchem<2.4.0` requires tensorflow v1 (>=1.14)


+3 −3
Original line number Diff line number Diff line
@@ -34,9 +34,9 @@ pip install -r $PWD/requirements-test.txt

# Fixed packages
tensorflow=2.2.0
torch=1.5.0
torchvision=0.6.1
pyg_torch=1.5.0
torch=1.6.0
torchvision=0.7.0
pyg_torch=1.6.0

# Install TensorFlow dependencies
pip install tensorflow==$tensorflow tensorflow-probability