NeurAI ：高性能神经科学计算库#
NeurAI 简介#
NeurAI 是一个基于 JAX 构建的高性能神经科学计算库，专为脉冲神经网络和机器学习研究而设计。有关 NeurAI 的文档可以在 NeurAI Documentation 上找到。
安装方法#
安装前需要先配置pip的索引。
# 首先创建一个pip的配置目录：
mkdir -p ~/.config/pip
# 然后编辑pip的配置文件：
vim ~/.config/pip/pip.conf
pip.conf 文件内容如下：
[global]
index-url = https://mirrors.aliyun.com/pypi/simple/
[install]
extra-index-url = https://nexus.nanhubrain.com/repository/pypi-hosted/simple/
平台 |
命令 |
|---|---|
APU |
|
CPU |
|
GPU (CUDA 11.0) |
|
GPU (CUDA 12.0) |
|
- 注意：
如果环境已安装有 cuda，安装 NeurAI 时，可以使用
neurai[cuda11_local]或者是neurai[cuda12_local]，这表明安装 NeurAI 时，不会安装 nvidia 相关的包。 详细安装指南请参考 安装指南。