daniilidis-group / neural_renderer

A PyTorch port of the Neural 3D Mesh Renderer
Other
1.12k stars 248 forks source link

tests: Add running to README; relax silhouette tests to use IoU #115

Open EricCousineau-TRI opened 2 years ago

EricCousineau-TRI commented 2 years ago

Just did some quick test fixes

Tested this against torch==1.7.1+cu110 on my machine, using @adambielski's fix in #86

EricCousineau-TRI commented 2 years ago

My full transcript on Ubuntu 18.04:

cuda-11.2-setup ()
{
    export CUDA_ROOT=/usr/local/cuda-11.2;
    export CUDA_INC_DIR=${CUDA_ROOT}/include;
    export PATH=${CUDA_ROOT}/bin:${PATH};
    export LD_LIBRARY_PATH=${CUDA_ROOT}/lib64:${LD_LIBRARY_PATH}
}

python3 -m venv ./venv
source ./venv/bin/activate
pip install -U pip wheel

pip install torch==1.7.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html
cuda-11.2-setup  # Mixing CUDA 11.0 and 11.2, but meh
pip install -e .

pip install scikit-image

cd ./tests
python3 -m unittest discover -t .