首页 > 其他分享 >kaggle上的jax框架的环境配置(TPU版本)

kaggle上的jax框架的环境配置(TPU版本)

时间:2024-01-18 21:12:44浏览次数:34  
标签:0.2 jax 0.1 nvidia kaggle TPU 3.0 tensorflow cu12

导出时间:

2024-01-18 21:00:37 星期四



python版本:

Python 3.10.13



absl-py==1.4.0
accelerate==0.25.0
aiofiles==22.1.0
aiosqlite==0.19.0
anyio==4.2.0
argon2-cffi==23.1.0
argon2-cffi-bindings==21.2.0
array-record==0.5.0
arrow==1.3.0
astroid==3.0.2
asttokens==2.4.1
astunparse==1.6.3
attrs==23.2.0
audioread==3.0.1
autopep8==2.0.4
Babel==2.14.0
beautifulsoup4==4.12.2
bleach==6.1.0
cachetools==5.3.2
certifi==2023.11.17
cffi==1.16.0
charset-normalizer==3.3.2
chex==0.1.85
click==8.1.7
cloud-tpu-client==0.10
cloudpickle==3.0.0
comm==0.2.1
contourpy==1.2.0
cycler==0.12.1
debugpy==1.8.0
decorator==5.1.1
defusedxml==0.7.1
diffusers==0.25.0
dill==0.3.7
distrax==0.1.5
dm-haiku @ git+https://github.com/deepmind/dm-haiku@6339353cab51d3d71f172d1bbe2d216c33ce09a4
dm-tree==0.1.8
docstring-to-markdown==0.13
entrypoints==0.4
etils==1.6.0
exceptiongroup==1.2.0
executing==2.0.1
fastjsonschema==2.19.1
filelock==3.13.1
flake8==6.1.0
flatbuffers==23.5.26
flax==0.7.4
fonttools==4.47.0
fqdn==1.5.1
fsspec==2023.12.2
funcsigs==1.0.2
gast==0.5.4
gin-config==0.5.0
google-api-core==1.34.0
google-api-python-client==1.8.0
google-auth==2.26.1
google-auth-httplib2==0.2.0
google-auth-oauthlib==1.2.0
google-pasta==0.2.0
googleapis-common-protos==1.62.0
grpcio==1.60.0
gym==0.26.2
gym-notices==0.0.8
h5py==3.10.0
httplib2==0.22.0
huggingface-hub==0.20.2
idna==3.6
importlib-metadata==7.0.1
importlib-resources==6.1.1
ipykernel==6.28.0
ipython==8.20.0
ipython-genutils==0.2.0
isoduration==20.11.0
isort==5.13.2
jax==0.4.17
jaxlib==0.4.17
jedi==0.19.1
Jinja2==3.1.2
jmp==0.0.4
joblib==1.3.2
jraph==0.0.6.dev0
json5==0.9.14
jsonpointer==2.4
jsonschema==4.20.0
jsonschema-specifications==2023.12.1
jupyter-events==0.9.0
jupyter-lsp==1.5.1
jupyter-ydoc==0.2.5
jupyter_client==7.4.9
jupyter_core==5.7.1
jupyter_server==2.12.3
jupyter_server_fileid==0.9.1
jupyter_server_terminals==0.5.1
jupyter_server_ydoc==0.8.0
jupyterlab==3.6.6
jupyterlab_pygments==0.3.0
jupyterlab_server==2.25.2
kagglehub==0.1.5
keras==3.0.2
keras-cv==0.8.1
keras-nlp==0.7.0
kiwisolver==1.4.5
lazy_loader==0.3
libclang==16.0.6
librosa==0.10.1
libtpu-nightly==0.1.dev20231003
llvmlite==0.41.1
Markdown==3.5.1
markdown-it-py==3.0.0
MarkupSafe==2.1.3
matplotlib==3.8.2
matplotlib-inline==0.1.6
mccabe==0.7.0
mdurl==0.1.2
mistune==3.0.2
ml-dtypes==0.2.0
mpmath==1.3.0
msgpack==1.0.7
nbclassic==1.0.0
nbclient==0.9.0
nbconvert==7.14.0
nbformat==5.9.2
nest-asyncio==1.5.8
networkx==3.2.1
notebook==6.5.6
notebook_shim==0.2.3
numba==0.58.1
numpy==1.26.3
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==8.9.2.26
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.18.1
nvidia-nvjitlink-cu12==12.3.101
nvidia-nvtx-cu12==12.1.105
oauth2client==4.1.3
oauthlib==3.2.2
opencv-python-headless==4.9.0.80
opt-einsum==3.3.0
optax==0.1.7
orbax-checkpoint==0.4.4
overrides==7.4.0
packaging==23.2
pandas==2.1.4
pandocfilters==1.5.0
papermill==2.5.0
parso==0.8.3
pexpect==4.9.0
pillow==10.2.0
platformdirs==4.1.0
pluggy==1.3.0
pooch==1.8.0
prometheus-client==0.19.0
promise==2.3
prompt-toolkit==3.0.43
protobuf==3.20.3
psutil==5.9.7
ptyprocess==0.7.0
pure-eval==0.2.2
pyasn1==0.5.1
pyasn1-modules==0.3.0
pycodestyle==2.11.1
pycparser==2.21
pydocstyle==6.3.0
pyflakes==3.1.0
Pygments==2.17.2
pylint==3.0.3
pyparsing==3.1.1
python-dateutil==2.8.2
python-json-logger==2.0.7
python-lsp-jsonrpc==1.1.2
python-lsp-server==1.9.0
pytoolconfig==1.2.6
pytz==2023.3.post1
PyYAML==6.0.1
pyzmq==24.0.1
referencing==0.32.1
regex==2023.12.25
requests==2.31.0
requests-oauthlib==1.3.1
rfc3339-validator==0.1.4
rfc3986-validator==0.1.1
rich==13.7.0
rope==1.11.0
rpds-py==0.16.2
rsa==4.9
safetensors==0.4.1
scikit-learn==1.3.2
scipy==1.11.4
Send2Trash==1.8.2
six==1.16.0
sniffio==1.3.0
snowballstemmer==2.2.0
soundfile==0.12.1
soupsieve==2.5
soxr==0.3.7
stack-data==0.6.3
sympy==1.12
tabulate==0.9.0
tenacity==8.2.3
tensorboard==2.15.1
tensorboard-data-server==0.7.2
tensorflow @ https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/tensorflow/tf-2.15.0/tensorflow-2.15.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
tensorflow-addons==0.23.0
tensorflow-datasets==4.9.4
tensorflow-estimator==2.15.0
tensorflow-hub==0.15.0
tensorflow-io==0.35.0
tensorflow-io-gcs-filesystem==0.35.0
tensorflow-metadata==1.14.0
tensorflow-probability==0.23.0
tensorflow-text==2.15.0
tensorstore==0.1.45
termcolor==2.4.0
terminado==0.18.0
threadpoolctl==3.2.0
tinycss2==1.2.1
tokenizers==0.15.0
toml==0.10.2
tomli==2.0.1
tomlkit==0.12.3
toolz==0.12.0
torch==2.1.0
torch-xla @ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.1.0+libtpu-cp310-cp310-manylinux_2_28_x86_64.whl
torchaudio==2.1.0
torchdata==0.7.0
torchtext==0.16.0
torchvision==0.16.0
tornado==6.4
tqdm==4.66.1
traitlets==5.14.1
transformers==4.36.2
trax==1.4.1
triton==2.1.0
typeguard==2.13.3
types-python-dateutil==2.8.19.20240106
typing_extensions==4.9.0
tzdata==2023.4
ujson==5.9.0
uri-template==1.3.0
uritemplate==3.0.1
urllib3==2.1.0
wcwidth==0.2.13
webcolors==1.13
webencodings==0.5.1
websocket-client==1.7.0
Werkzeug==3.0.1
whatthepatch==1.0.5
wrapt==1.14.1
y-py==0.6.2
yapf==0.40.2
ypy-websocket==0.8.4
zipp==3.17.0

标签:0.2,jax,0.1,nvidia,kaggle,TPU,3.0,tensorflow,cu12
From: https://www.cnblogs.com/devilmaycry812839668/p/17973406

相关文章

  • Google的jax框架在TPU上的循环控制 —— 向量计算设备的循环结构控制
    相关:https://jax.readthedocs.io/en/latest/pallas/tpu.html向量计算设备,如:GPU、TPU等,都是通过向量计算来进行加速的,因此在这类设备中进行向量计算的计算单元是成百上千的,但是进行结构控制的电路单元比较少,可以基本认为在向量设备中进行流程控制是标量的,而不是向量的,也就是说......
  • jax框架的 Pallas 方式的GPU扩展不可用
    说下深度学习框架的GPU扩展功能的部分,也就是使用个人定制化的GPU代码编写方式来为深度学习框架做扩展。深度学习框架本身就是一种对GPU功能的一种封装和调用,但是由于太high-level,因此就会摒弃掉一些原有的GPU底层的编程功能,为此可以使用GPU原始功能的代码来为深度学习编写扩展函......
  • $.ajax()方法详解
    1.url要求为String类型的参数,(默认为当前页地址)发送请求的地址。2.type要求为String类型的参数,请求方式(post或get)默认为get。注意其他http请求方法,例如put和delete也可以使用,但仅部分浏览器支持。3.timeout要求为Number类型的参数,设置请求超时时间(毫秒)。此设置将覆盖$.ajaxSetup()方......
  • Ajax注册表单用户名实时验证
    Ajax注册表单用户名实时验证 1.1.1摘要很多时候在网站上注册时,我们会发现,注册表单通常需要检查用户名和电子邮件地址的可用性;从而确保用户之间不拥有相同的用户名和电子邮件地址;一些网站喜欢在用户提交填写的用户信息时,做信息可用性的检查,而一些网站会做实时的用户名和电子......
  • ajax实现文件上传
    ajax上传文件上传文件我一般采用form表单上传,form简单form上传文件<formid="jsSubmit"method="post"action="url"enctype="multipart/form-data"><inputtype="file"name="file"/></form><......
  • Java中的InputStream和OutputStream详解
    引言在Java编程中,处理输入输出是日常任务的一部分,而流(Stream)是实现输入输出的核心概念。在JavaI/OAPI中,InputStream和OutputStream是所有字节流类的基础。本文将详细介绍这两个类及其在Java中的应用。什么是InputStream和OutputStream?InputStream是JavaI/O库中的一个抽象类,它......
  • 在Java HttpURLConnection库上编写一个下载程序下载美图秀秀的图片
    在JavaHttpURLConnection库上编写一个下载程序,该下载程序使用Kotlin下载https://xiuxiu.meitu.com/的内容,代码必须使用以下代码:代理主机:www.duoip.cn,代理端口:8000。importjava.io.BufferedReaderimportjava.io.InputStreamReaderimportjava.net.HttpURLConnectionimportjav......
  • Ajax异步交互技术
    在HTTP协议下,前后端交互有两种技术,一是同步,二是异步Ajax异步交互技术简介“AsynchronousJavascriptAndXML”(异步JavaScript和XML),是指一种创建交互式、快速动态网页应用的网页开发技术,无需重新加载整个网页的情况下,能够更新部分网页的技术。通过在后台与服务器进行少量数据......
  • Ajax异步交互技术
    在HTTP协议下,前后端交互有两种技术,一是同步,二是异步Ajax异步交互技术简介“AsynchronousJavascriptAndXML”(异步JavaScript和XML),是指一种创建交互式、快速动态网页应用的网页开发技术,无需重新加载整个网页的情况下,能够更新部分网页的技术。通过在后台与服务器进行少量数据......
  • Ajax异步交互技术
    在HTTP协议下,前后端交互有两种技术,一是同步,二是异步Ajax异步交互技术简介“AsynchronousJavascriptAndXML”(异步JavaScript和XML),是指一种创建交互式、快速动态网页应用的网页开发技术,无需重新加载整个网页的情况下,能够更新部分网页的技术。通过在后台与服务器进行少量......