【Python/Pytorch】-- 创建 tiny-cuda-nn 环境

电科_银尘2026-01-04 10:48

创建 tiny-cuda-nn 环境

文章目录

[01 第一步](#01 第一步)
[02 第二步](#02 第二步)
[03 第三步](#03 第三步)
[04 第四步](#04 第四步)

01 第一步

查看服务器的cuda版本和显卡类型

python 复制代码

查看当前cuda版本的指令 nvcc --version
查看gpu相关信息nividia-smi

02 第二步

根据服务器cuda的版本和显卡类型，安装对应的python版本（3.10）

python 复制代码

conda create --prefix /path/to/your/enviroment python=3.9
conda activate /path/to/your/enviroment

因为该服务器的cuda版本是cuda 11.8，显卡为4090Ti，因此最后的安装为：

python 复制代码

方法1：直接指定版本（推荐）
pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 
--index-url https://download.pytorch.org/whl/cu118
方法2：如果网络不行，使用国内镜像
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 
--index-url https://mirrors.aliyun.com/pypi/simple 
--extra-index-url https://download.pytorch.org/whl/cu118

03 第三步

进入tiny-cuda-nn的安装包，安装指示安装包https://github.com/NVlabs/tiny-cuda-nn

python 复制代码

cd /path/to/your/enviroment/tiny-cuda-nn/bindings/torch
tiny-cuda-nn/bindings/torch$ python setup.py install

04 第四步

安装项目依赖的其他包，遇到的几个问题的解决方法记录：

1、medutils 和 yaml 直接安装失败，可以换成安装medutils-mri 和 yaml-1-3 、yaml-acl，中间还要安装正确的h5py的版本