我正在将 AWS P2 gpu 实例与 Tesla K80 GPU 和 Ubuntu 16.04 LTS 一起使用。为 CPU 安装 Tensorflow 很容易:
pip install tensorflow
但是为 GPU 安装 Tensorflow 不起作用:
pip install tensorflow-gpu
自动安装所有推荐的 ubuntu 驱动程序(包括 nvidia 驱动程序):
sudo apt-get purge nvidia*
sudo ubuntu-drivers autoinstall
系统将要求您设置密码以验证安装是否为正版。设置该密码后,重新启动计算机。然后不要在正常模式下启动,进入 MOK 模式并输入该密码作为密钥。重新启动(重新启动)后,通过以下方式检查驱动程序是否安装成功:
nvidia-smi
然后安装 Anaconda链接,它将处理一切。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句