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

导出时间：

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