Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
N
NiuTrans.Tensor
概览
Overview
Details
Activity
Cycle Analytics
版本库
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
问题
0
Issues
0
列表
Board
标记
里程碑
合并请求
0
Merge Requests
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
Snippets
成员
Collapse sidebar
Close sidebar
活动
图像
聊天
创建新问题
作业
提交
Issue Boards
Open sidebar
杨迪
NiuTrans.Tensor
Commits
1bf5cc90
Commit
1bf5cc90
authored
Apr 17, 2020
by
liyinqiao
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Update the manual.
Update the minimum version of CUDA from 9.0 to 9.2.
parent
22cc1218
显示空白字符变更
内嵌
并排
正在显示
2 个修改的文件
包含
4 行增加
和
4 行删除
+4
-4
Makefile
+1
-1
doc/manual.md
+3
-3
没有找到文件。
Makefile
查看文件 @
1bf5cc90
...
...
@@ -18,7 +18,7 @@ OnMac = 0
# 1 - use GPU
USE_CUDA
=
0
# modify this path if neccessary
CUDA_ROOT
=
/usr/local/cuda
-9.0
CUDA_ROOT
=
/usr/local/cuda
CUDA_LIB_DIR
=
$(CUDA_ROOT)
/lib64
CUDA_INCLUDE
=
$(CUDA_ROOT)
/include
...
...
doc/manual.md
查看文件 @
1bf5cc90
...
...
@@ -16,7 +16,7 @@ NiuTensor工具包的安装方法目前支持CMake(跨平台:支持Windows
在开始创建您的项目并使用NiuTensor工具包时,需要
**注意**
的是:
*
所创建项目如在CPU上运行,我们的系统支持高性能的数学运算库,推荐安装
[
MKL
](
https://software.intel.com/en-us/mkl
)
或
[
OpenBLAS
](
http://www.openblas.net/
)
(目前CMake方式不支持MKL和OpenBLAS,如希望使用上述运算库,建议使用Visual Studio或Makefile的方式进行编译,后续CMake将提供对其的完整支持)。
*
所创建项目如需在GPU上运行,需安装
[
CUDA
](
https://developer.nvidia.com/cuda-downloads
)
,CUDA版本需求为9.
0
及以上,CUDA工具为创建高性能GPU加速应用程序提供了开发环境。
*
所创建项目如需在GPU上运行,需安装
[
CUDA
](
https://developer.nvidia.com/cuda-downloads
)
,CUDA版本需求为9.
2
及以上,CUDA工具为创建高性能GPU加速应用程序提供了开发环境。
### 编译文件的修改
...
...
@@ -28,7 +28,7 @@ NiuTensor工具包的安装方法目前支持CMake(跨平台:支持Windows
-
打开CMakeLists.txt文件对其进行编辑。
-
操作系统设置:若NiuTensor编译环境为Windows,则在
`set(ON_WINDOWS 0)`
中将
`ON_WINDOWS`
的值置为1;若编译环境为Linux或macOS,则将
`ON_WINDOWS`
的值置为0。
-
编译设备设置:若希望在CPU环境下编译使用NiuTensor工具包,则将
`set(USE_CUDA 0)`
中的
`USE_CUDA`
置为0即可;若希望在GPU环境下使用,则需将
`USE_CUDA`
置为1,同时在
`set(CUDA_VERSION 9.
0
)`
中设置CUDA版本号,在
`set(CUDA_TOOLKIT_ROOT_DIR "/usr/local/cuda")`
中设置CUDA工具安装的根目录。
-
编译设备设置:若希望在CPU环境下编译使用NiuTensor工具包,则将
`set(USE_CUDA 0)`
中的
`USE_CUDA`
置为0即可;若希望在GPU环境下使用,则需将
`USE_CUDA`
置为1,同时在
`set(CUDA_VERSION 9.
2
)`
中设置CUDA版本号,在
`set(CUDA_TOOLKIT_ROOT_DIR "/usr/local/cuda")`
中设置CUDA工具安装的根目录。
#### Makefile
...
...
@@ -36,7 +36,7 @@ NiuTensor工具包的安装方法目前支持CMake(跨平台:支持Windows
-
打开Makefile文件对其进行编辑。
-
操作系统设置:若NiuTensor编译环境为Windows或Linux,则在
`OnMac = 0`
中将
`OnMac`
的值置为0;若编译环境为macOS,则将
`OnMac`
的值置为1。
-
编译设备设置:若希望在CPU环境下编译使用NiuTensor工具包,则将
`USE_CUDA = 0`
中的
`USE_CUDA`
置为0即可;若希望在GPU环境下使用,则需将
`USE_CUDA`
置为1,同时在
`CUDA_ROOT = /usr/local/cuda
-9.0
`
中设置CUDA工具安装的根目录。
-
编译设备设置:若希望在CPU环境下编译使用NiuTensor工具包,则将
`USE_CUDA = 0`
中的
`USE_CUDA`
置为0即可;若希望在GPU环境下使用,则需将
`USE_CUDA`
置为1,同时在
`CUDA_ROOT = /usr/local/cuda`
中设置CUDA工具安装的根目录。
-
编译内容设置:若希望生成NiuTensor工具包的动态链接库,则将
`dll = 0`
中的
`dll`
置为1即可;若无需编译动态链接库,则将其置为0。
### 工具包的安装
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论