anaconda,cuda,torch,lightning的安装

2023-11-19

本博客仅作为初学者参考使用,汇总了多位大牛的博客,如有侵权请联系我删除

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缓存

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

anaconda,cuda,torch,lightning的安装 的相关文章

随机推荐