本博客仅作为初学者参考使用,汇总了多位大牛的博客,如有侵权请联系我删除
anaconda,cuda,torch,lightning的安装
- 1、Anaconda
- 2、cuda
- 3、pytorch
- 4、lightning
- 5、解决pip执行后导致C盘空间变小问题
1、Anaconda
作用:
1、可创建python包的虚拟环境,方便管理;
2、包含了许多常用的包,方便开发人员使用
安装请参考:参考文档-Anaconda
注意版本选择
2、cuda
作用:加快程序运行速度
安装请参考:参考文档-cuda
注意环境变量配置以及版本选择
命令行运行nvidia-smi
,可找到如下图所示红框中对应的版本信息
3、pytorch
作用:以Python 优先的深度学习框架,不仅能够实现强大的GPU 加速,同时还支持动态神经网络。
命令行运行:
pip install --user torch==1.10.0 -f https://download.pytorch.org/whl/torch_stable.html
注意:我安装的torch-1.10.0 与 cuda 12.2 适配,但是如果torch版本再升高可能就不匹配了,可尝试更高版本的torch
4、lightning
作用:作为pytorch的一个拓展架构,可以减少很多与数据处理以及模型搭建无关的代码,增加工程效率。
pip install pytorch_lightning==1.8.0
注意:lightning的版本同样要和torch版本对应。
版本对应参考:
参考文档-lightning
5、解决pip执行后导致C盘空间变小问题
实际上就是缓存问题导致的,pip下载生成的缓存的默认位置是C盘
参考文档-pip缓存