目录
https://github.com/erksch/waymo-open-dataset-viewer/tree/master
# 安装 tensorflow (需要注意tensorflow与cuda版本对应,我这里使用cuda 11.0版本)
pip install tensorflow==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
#安装 pytorch (pytorch-lightning 是否安装自行选择)
pip install pytorch-lightning==1.4.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
# 测试gpu是否可用(返回 gpu 信息和 True 则表示安装正确)
python
>>> import tensorflow as tf
>>> tf.config.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
>>> import torch
>>> torch.cuda.is_available()
True
>>> quit()
# 安装 waymo-open-dataset 需要与tensorflow版本对应
pip install waymo-open-dataset-tf-2-4-0
waymo-open-dataset 没找到,pypi有安装包,只支持linux 系统。
https://blog.csdn.net/weixin_50232758/article/details/132260047
?
https://github.com/waymo-research/waymo-open-dataset/tree/master/docs