【高级深度学习框架】lightning的使用记录
目录
- 问题1 :如何安装
- 解决方案
问题1 :如何安装
解决方案
来源:官网
有两种方式:
(1)pip安装。python -m pip install lightning
(2)conda安装。conda install lightning
可以不用加 -c conda-forge 参数
安装方式的选择,根据自己环境中pytorch的安装方式决定。 即,
-
若pytorch使用pip进行安装,则采用pip安装lightning;
-
若pytorch使用conda进行安装,则采用conda安装lightning。
问:如何查看自己的pytorch是使用何种方式安装?答:使用以下代码查看pytorch的位置:1 import torch2 print(torch.__file__)若打印结果为/usr/local/lib/xx...,则是采用pip进行安装若打印结果为 /anaconda3/envs/xxx/lib/xx...,则是采用conda进行安装
附: 若在安装过程中,出现问题,则需要检查一下自己的环境是否有问题。例如,网上有人说无法安装是因为torch版本啥的问题,分析是有误的。只要你的环境和pytorch是在矩阵中是存在对应关系,则一定是可以顺利安装的。