置信不少人在“炼丹”进程中,光是安装或更新下面这“几大位”时就阅历了一段"血泪史"吧:
能不能挽救一下?
能,如今你运用Lambda Stack,就能完成一行命令打包安装或更新好TensorFlow与PyTorch等一切“炼丹”工具,包括一切的依赖项!
那么擦干眼泪,学起来?
Lambda Stack与安装首先来了解一下Lambda Stack是什么。
这就是由Lambda创立的一个Debian PPA (团体软件包存档)。
目前,外面为你提供了这些工具的软件包:
TensorFlow v2.4.1
PyTorch v1.8.0
CUDA v11.1
cuDNN v8.0.5
依赖项及其他框架,如Caffe、Theano
然后大家经过系统的APT工具也就是sudo apt-get install/update命令,就可以很方便地下载外面的各种.deb包了。
首先,反省一下系统要求:
NVIDIA GPU (如RTX 3090, 3080, 3070, 2080 Ti, A6000, Quadro RTX 8000)
Ubuntu 20.04 LTS
接上去,假设你是desktop版Linux系统,就请执行以下命令:
server版系统请执行:
一切正常以后,重启一下机子,就可以运用了:
假设上述软件有更新,很复杂,只需下面这一行命令就ok:
ps.更新后也要重启。
是不是方便多了。
下面摘取了Reddit上网友针对以上安装进程的一些疑问,以及官方人员的回复。
Q&A1、与Conda有何不同?
Conda可以给你装CUDA工具包,但不会装NVIDIA驱动顺序;而Lambda Stack都能安装。
此外,还可将Lambda Stack与pip、虚拟环境一同运用
2、能否组合特定版本,比如CUDA 9.2 + PyTorch 1.5?
不能,它提供的都是CUDA、PyTorch、Tensorflow以及NVIDIA驱动的最新兼容版本,混搭不行。
3、可以在 Amazon Sagemaker机器上运转吗?
可以,任何机器上都能收费安装。
4、安装包大约多大?我只要一个小的SSD,我家带宽也有限。
大约在1-6GB之间,确切数字“我”不记得了;安装应该还挺快的。
最后,官方人员表示他们行将发布一个视频,解说如何将Lambda Stack与Docker、Nvidia-Container-Toolkit(前Nvidia-Docker)一同运用。敬请等候吧。
官方教程:
https://lambdalabs.com/blog/install-tensorflow-and-pytorch-on-rtx-30-series/
软件源:
https://lambdalabs.com/static/misc/lambda-stack-repo.deb
【编辑引荐】
七个用于图形深度学习的开源代码库
2021 年 6 月头号恶意软件:Trickbot 依然位居榜首
完成数据常青 XSKY星斗天合发布企业级软件定义存储V5系列
企业如何提高数据库安全?清点这11种工具和技术
网络钓鱼盯上whatsapp等即时通讯软件 卡巴斯基给出安全建议
(责任编辑:admin)