TX2(秒算2)安裝Tensorflow採坑記錄

環境: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/

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章