本机系统 ubuntu 22.04 x86_64
GPU:RTX3080TI 16G
驱动程序版本 nvidia 530
CUDA版本 11.8
cuda 下载官网
(建议看完之后在进行操作)
- 使用ubuntu本机自带的附加驱动 选择专用的530驱动(nvidia-smi测试)
- 安装cuda local版本 .run的版本
a. chmode +x cuda__linux.run
b. (时间较长的等待后) 选择不安装驱动程序 路径选择非/usr/local下
ⅰ. 第一个出现的页面是选择安装那些不安装那些,我是不安装驱动,所以选择第一个X 回车一下取消[ ]中的X
ⅱ. 第二页面是输入 accept 同意,等待即可
c. nvcc 安装(这个安装完成之后,nvcc --version命令显示cuda是11.5,不知道为什么降低了cuda版本)
d. 重新安装cuda11.8版本,设置环境变量 但是/usr/local下还是有个cuda
e. 重启 重新测试之后nvcc 显示cuda版本正常 - 安装torch和对应的cuda版本 选的是11.8的正常安装
- tiny-cuda-nn 的安装让我崩溃
a. 我是分别从github上 下载zip tiny-cuda-nn 下载zip fmt 下载zip cutlass, 将这两个替换到tiny-cuda-nn/ dependencies/下
b. cmake . -B build 、 cmake --build build --config RelWithDebInfo (一般版本对应上之后,第一个命令不会有什么报错,第二个命令慢一些也容易出问题,我执行的过程中没有出现问题,版本没有对应的时候一直报错)
c. cd tiny-cuda-nn/bindings/torch/ 执行 python setup.py install (执行第一次的时候报错了,也许不是报错,可能是警告,然后我切断了程序,又重新执行了,最后安装成功了) - python 测试 import tinycudann 没有报错就是安装成功了
主要是cuda版本和nvcc查询的版本一致,torch版本一致 应该就没什么问题