a | b/install.sh | ||
---|---|---|---|
1 | conda install -c anaconda cudnn=7.6.5 cudatoolkit=10.0.130 -y |
||
2 | conda install -c pytorch pytorch torchvision -y |
||
3 | conda install lightgbm numpy scikit-learn tqdm opencv matplotlib pandas pillow=6.2.1 scipy=1.1.0 -y |
||
4 | conda install -c conda-forge imgaug=0.3.0 tensorboardX -y |
||
5 | pip install keras==2.1.5 keras-applications==1.0.8 keras-preprocessing==1.1.0 keras-resnet==0.1.0 tensorflow-gpu==1.15.2 |