# 配置英伟达 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