Skip to content

Instantly share code, notes, and snippets.

@suddhu

suddhu/pytorch_minkowski_conda.sh

Last active September 13, 2022 14:18
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save suddhu/3621ac241452d1321ad7493b7029fd99 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save suddhu/3621ac241452d1321ad7493b7029fd99 to your computer and use it in GitHub Desktop.
Download ZIP
PyTorch + CUDA + MinkowskiEngine setup
Raw
pytorch_minkowski_conda.sh
conda install pytorch=1.8.1 torchvision cudatoolkit=10.1 -c pytorch -c conda-forge
export CXX=g++-7
export CUDA_HOME=/usr # verify CUDA location via $CUDA_HOME/nvcc --version
pip install -U git+https://github.com/NVIDIA/MinkowskiEngine -v --no-deps --install-option="--blas_include_dirs=${CONDA_PREFIX}/include" --install-option="--blas=openblas"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment