1. 程式人生 > 其它 >tensorflow-gpu == 2.0.0-beta1安裝說明

tensorflow-gpu == 2.0.0-beta1安裝說明

技術標籤:tensorflowpytorch

安裝步驟說明

  • 建立環境
conda create -n TF2.0 python=3.7
  • 啟用環境
conda activate TF2.0
  • 安裝cudatoolkit
conda install cudatoolkit=10.1
  • 安裝cudnn
conda install cudnn=7.6
  • 安裝tensorflow-gpu == 2.0.0-beta1
pip install tensorflow-gpu == 2.0.0-beta1 -i https://pypi.douban.com/simple