環境:jetpack3.3(cuda9.0)
步驟一:先裝pip
sudo apt-get install python-pip
可能需要升級,需要改pip的main()文件
sudo gedit /usr/bin/pip
可能遇到更新 pip 後出現 main 錯誤,將
from pip import main
if __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/