前言
参考:
【深度学习】【三维重建】windows11环境配置tiny-cuda-nn详细教程
nerfstudio搭建 win11踩坑记录之tinycudann
提示:以下是本篇文章正文内容,下面案例可供参考
一、安装方法
亲测好用版本:
Windows10
cuda 11.7
pytorch:2.0.1+cu117(注意一定是cuda版本)
python:3.9
visual studio 2022
提示:如果需要安装多版本cuda,一定要多看教程,谨慎!
二、安装步骤
1.安装所需软件
安装anaconda:anaconda官网
安装visual studio 2022:免费的社区版即可,参考Win10下Visual Studio 2019安装教程(图文)
安装cuda和cudnn:Win10 + CUDA11.7+pytorch手动安装-2.0版本
创建环境:
conda create -n py39 python=3.9
activate py39
安装pytorch:
安装代码如下(示例):
pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117
2.安装tinycudann
这部分主要参考大佬nerfstudio搭建 win11踩坑记录之tinycudann
代码如下(示例):
#此处路径根据自己的文件路径做调整
start "C:\Program Files\Microsoft Visual Studio\2022\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
之后在弹出的新的cmd中
pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
等一会
测试
python
import tinycudann