Commit 86d025b2 authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

adding libraries for doc fix

parent 185294a3

env.common.yml

+3 −0
Original line number Diff line number Diff line
@@ -29,3 +29,6 @@ dependencies: 
    - transformers==4.6.*

    - xgboost==1.*

    - git+https://github.com/samoturk/mol2vec

    - git+https://github.com/deepmind/dm-haiku

    - git+git://github.com/deepmind/optax.git

env.cpu.mac.yml

+1 −0
Original line number Diff line number Diff line
@@ -7,3 +7,4 @@ dependencies: 
    - torchvision==0.9.1

    - torch-scatter

    - torch-sparse

    - jax[cpu]

env.cpu.yml

+1 −0
Original line number Diff line number Diff line
@@ -7,3 +7,4 @@ dependencies: 
    - torchvision==0.9.1+cpu

    - torch-scatter

    - torch-sparse

    - jax[cpu]

env.gpu.yml

+2 −0
Original line number Diff line number Diff line
@@ -7,3 +7,5 @@ dependencies: 
    - torchvision==0.9.1+cu111

    - torch-scatter

    - torch-sparse

    - jax[cuda111]

CRA Git | Maintained and supported by SUSTech CRA and CCSE