Commit 8f7a6c75 authored by wangys_biolab's avatar wangys_biolab

Update README.md

parent d75d781a
Pipeline #9 canceled with stages
......@@ -13,6 +13,11 @@ prPred-DRLF is an open-source Python-based toolkit, which operates depending on
### **Install dependencies**
conda create -n prPred-DRLF python=3.7
conda activate prPred-DRLF
pip3 install -r requirements.txt
or
pip3 install joblib==1.0.1
......@@ -23,12 +28,27 @@ prPred-DRLF is an open-source Python-based toolkit, which operates depending on
pip3 install lightgbm-2.3.0
#### For python3.7
#### If you have GPU # CUDA 9.2
#### If you have GPU # CUDA 9.2
pip install torch==1.2.0 torchvision==0.4.0
#### If you have GPU CUDA 10.0
#### If you have GPU CUDA 10.0
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
**# CUDA 10.0**
conda install pytorch==1.3.1 torchvision==0.4.2 cudatoolkit=10.0 -c pytorch
#### CPU Only python3.7
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
**# CPU Only**
conda install pytorch==1.3.1 torchvision==0.4.2 cpuonly -c pytorch
### Input parameters
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment