-
Recommended to use Anaconda python distribution.
-
Download from here and install it.
-
Install python libraries:
conda install -y numpy=1.16.4 scipy=1.3 scikit-learn=0.21.2 h5py=2.9.0 -
Install PyTorch:
conda install -y pytorch torchvision cudatoolkit=10.0 -c pytorch