File size: 729 Bytes
b170003 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
#!/bin/sh
conda create -n discomat python=3.7.9
conda activate discomat
conda install -y numpy==1.20.3 pandas==1.2.4 scikit-learn=0.23.2
conda install -y pytorch==1.7.0 cudatoolkit=10.1 -c pytorch
conda install -c conda-forge fairseq
conda install -c anaconda ujson
conda install -c conda-forge msgpack-python
conda install redis-py
conda install -c anaconda h5py
whl="torch_scatter-2.0.7-cp37-cp37m-linux_x86_64.whl"
curl "https://data.pyg.org/whl/torch-1.7.0%2Bcu101/${whl}" --output $whl
pip install $whl
rm $whl
whl="dgl_cu101-0.7.1-cp37-cp37m-manylinux1_x86_64.whl"
curl "https://data.dgl.ai/wheels/dgl_cu101-0.7.1-cp37-cp37m-manylinux1_x86_64.whl" --output $whl
pip install $whl
rm $whl
pip install -r requirements.txt
|