pytorch_lightning 安装

2023-12-14 21:16:00

安装pytorch-lightning时一定注意自己的torch是pip安装还是conda安装,pytorch_lightning 安装方式要与torch的安装方式保持一致,否则也会导致你的torch版本被替换。

正确安装方式:

pip方式:

pip install pytorch-lightning==版本名

?conda方式(conda不需要指定版本号)

conda install pytorch-lightning -c conda-forge

文章来源:https://blog.csdn.net/weixin_43135178/article/details/134719948
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。