速装jax

最新推荐文章于 2025-06-24 19:39:53 发布

飞行codes

最新推荐文章于 2025-06-24 19:39:53 发布

阅读量1.8k

点赞数

CC 4.0 BY-SA版权

分类专栏： T型牌坊文章标签：人工智能

本文链接：https://blog.csdn.net/Arctic_Beacon/article/details/122543259

T型牌坊专栏收录该内容

45 篇文章

订阅专栏

本文介绍了如何安装和配置JAX及其相关库的过程，包括创建环境、安装JAX及依赖库、选择安装Colab以及补装IDE等内容，并提供了国内镜像源的设置方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

jax可以理解为一个函数库，具有求导数的函数。

1、建立一个环境

2、安装jax

pip install --upgrade jax jaxlib==0.1.74 -f https://storage.googleapis.com/jax-releases/jax_releases.html

3、选装colab

pip install google-colab

4、补装ide

pip install spyder -i https://pypi.tuna.tsinghua.edu.cn/simple

添加镜像

mkdir ~/.pip
vim ~/.pip/pip.conf

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

pip国内的一些镜像

阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/