自学内容网 自学内容网

关于torch、Pytorch_lightning、torchvision版本的问题

# CUDA 11.3
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

安装lightning

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pytorch-lightning==1.7.6

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchmetrics==0.9.3


原文地址:https://blog.csdn.net/Daniel_Singularity/article/details/135346039

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!