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/