Deep Global Registration 代码环境配置(rtx3090+python3.8+cuda11.1+pytorch1.7+MinkowskiEngine0.5.1)

前言

踩过的坑:因为rtx3090最低算力是8.6,似乎不支持过低版本的cuda。试过pytorch1.7.0+cuda11.0,但会报错,由于cuda11.0 支持的最高算力达不到rtx最低的要求。但配置pytorch1.8时DGR的代码运行时会报错。
对于没有这个烦恼的还是推荐安装python3.6+cuda10.2+pytorch1.6+MinkowskiEngine0.4.3, 这个配置无需改动代码可直接跑通。

The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

1. 安装cuda和对应版本的cudnn并配置环境变量

  • 参考: 非root用户配置cuda环境
  • cuda 11.1 官网获取下载地址
    根据自己的环境选择runfile:

你可能感兴趣的:(pytorch,深度学习,神经网络)