The code is developed using Python 3.6 with PyTorch 1.0.0. The code is developed and tested using 2 GPU cards.
-
Clone this repo.
$ git clone https://github.com/charlesCXK/TorchSSC.git $ cd TorchSSC -
Install dependencies.
(1) Create a conda environment:
$ conda env create -f ssc.yaml $ conda activate ssc
(2) Install apex 0.1(needs CUDA)
$ cd ./furnace/apex $ python setup.py install --cpp_ext --cuda_ext