forked from jiuyuan/InfiniTensor
35 lines
1.2 KiB
Bash
35 lines
1.2 KiB
Bash
# 配置英伟达 CUDA 的 HOME 路径，请注意安装 CUDA Toolkit, CUDNN 并将路径配置到下述环境变量。
|
|
export CUDA_HOME=/PATH/TO/YOUR/CUDA/HOME
|
|
export CUDNN_HOME=/PATH/TO/YOUR/CUDNN/HOME
|
|
|
|
# 配置寒武纪 BANG 的 HOME 路径，请注意 /usr/local/neuware 是寒武纪软件栈建议的，同时也是默认的安装路径。
|
|
# 如若用户有其他的路径安装方式，请自行配置正确的路径。
|
|
# 这里是 neuware 目录下一个可能的结构图，请参考。
|
|
# .
|
|
# ├── bin
|
|
# ├── cmake
|
|
# ├── data
|
|
# ├── edge
|
|
# ├── include
|
|
# ├── lib
|
|
# ├── lib64
|
|
# ├── LICENSE
|
|
# ├── mlvm
|
|
# ├── README
|
|
# ├── samples
|
|
# ├── share
|
|
# └── version.txt
|
|
export NEUWARE_HOME=/usr/local/neuware
|
|
|
|
# 配置昆仑芯 XPU 的 HOME 路径，请注意 /usr/local/xpu 是昆仑芯软件栈提供的软件包路径。
|
|
# 如若用户有其他的路径安装方式，请自行配置正确的路径。
|
|
# 这里是 xpu 目录下一个可能的结构图，请参考。
|
|
# .
|
|
# ├── bin
|
|
# ├── include
|
|
# ├── lib64
|
|
# ├── tools
|
|
# ├── version
|
|
# └── XTDK
|
|
export XPU_HOME=/usr/local/xpu
|