本文共 608 字,大约阅读时间需要 2 分钟。
环境:jetpack3.3(cuda9.0)
步骤一:先装pip
sudo apt-get install python-pip
可能需要升级,需要改pip的main()文件
sudo gedit /usr/bin/pip
可能遇到更新 pip 后出现 main 错误,将
from pip import mainif __name__ == '__main__': sys.exit(main())
改为:
from pip import __main__if __name__ == '__main__': sys.exit(__main__._main())
步骤二:setuptools:
sudo pip install setuptools==39.1.0
步骤三:enum34:
sudo pip install --ignore-installed enum34
步骤四:tensorflow:
sudo pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp33 tensorflow-gpu
可能会遇到问题:failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED
sudo rm -rf ~/.nv/
转载地址:http://vnisn.baihongyu.com/