按照官网的方法直接pip .安装的,运行例子发现:
如上图,出现了一堆编译错误,主要是最后一句nvcc fatal : Value ‘c++17’ is not defined for option ‘std’,看了一下/usr/bin/nvcc,发现是10.1版本的cuda,不支持c++17。
然后看了一下机器上装了cuda-11.3在/usr/local下面,因为是第一次用这个机器,用户的环境变量里没有加新版本的cuda,于是在~/.bashrc里加入下面两行:
export PATH="/usr/local/cuda-11.3/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH"