近日,新入一台RTX3080的服务器,目前好像还没办法很方便地在 RTX 30 系列 GPU上通过 pip/conda 安装 TensorFlow 或 PyTorch。...因为这些 GPU 需要 CUDA 11.1,而当前主流的 TensorFlow/PyTorch 版本不是针对 CUDA 11.1 编译的。...简介
Lambda Stack,提供一行代码安装并可以管理升级,包含:Pythorch、TensorFlow、CUDA、cuDNN和NVIDIA驱动程序,它与Ubuntu 20.04 LTS,18.04...: Lambda Stack Dockerfiles
包含的深度学习框架: TensorFlow, Keras, PyTorch, Caffe, Caffe 2
包含的GPU 软件: CUDA, cuDNN...3090, 3080, 3070, 2080 Ti, Quadro RTX 8000)
Ubuntu 20.04, 18.04, and 16.04(桌面版和服务器版本都可以)
使用说明
如果是桌面版的