forked from jiuyuan/InfiniTensor
fix code
This commit is contained in:
parent
3caeeabacc
commit
3c1359b166
|
@ -36,7 +36,16 @@
|
|||
> > [CUDA Toolkit 安装](https://developer.nvidia.com/cuda-toolkit),
|
||||
> > [Cudnn 安装](https://developer.nvidia.com/rdp/cudnn-download),
|
||||
> > [Cublas 安装](https://developer.nvidia.com/cublas),
|
||||
> > 安装完成后请进行相应的环境变量配置,将可执行文件目录与库目录添加到操作系统识别的路径中,
|
||||
> > 安装完成后请进行相应的环境变量配置,将可执行文件目录与库目录添加到操作系统识别的路径中,例如
|
||||
> > ```bash
|
||||
> > # 将如下内容写入到你的 bashrc 文件并 source 该文件
|
||||
> > export CUDA_HOME="/PATH/TO/YOUR/CUDA_HOME"
|
||||
> > export CUDNN_HOME="/PATH/TO/YOUR/CUDNN_HOME"
|
||||
> > export PATH="${CUDA_HOME}/bin:${PATH}"
|
||||
> > export LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
|
||||
> > # 如您不方便将上述环境变量配置到 bashrc 文件中进行长期使用,你也可以在我们提供的 env.sh 文件中进行正确配置并激活,作为临时使用
|
||||
> > source env.sh
|
||||
> > ```
|
||||
我们强烈建议您规范安装,统一到一个目录下,以免不必要的麻烦。
|
||||
|
||||
> 如您的第三方加速卡为寒武纪 MLU,请参考寒武纪官方文档进行:
|
||||
|
|
2
env.sh
2
env.sh
|
@ -1,6 +1,8 @@
|
|||
# 配置英伟达 CUDA 的 HOME 路径,请注意安装 CUDA Toolkit, CUDNN 并将路径配置到下述环境变量。
|
||||
export CUDA_HOME=/PATH/TO/YOUR/CUDA/HOME
|
||||
export CUDNN_HOME=/PATH/TO/YOUR/CUDNN/HOME
|
||||
export PATH="${CUDA_HOME}/bin:${PATH}"
|
||||
export LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
|
||||
|
||||
# 配置寒武纪 BANG 的 HOME 路径,请注意 /usr/local/neuware 是寒武纪软件栈建议的,同时也是默认的安装路径。
|
||||
# 如若用户有其他的路径安装方式,请自行配置正确的路径。
|
||||
|
|
Loading…
Reference in New Issue