參考文章:
1. 安裝CUDA工具包10.1豺撑、cuDNN
NVIDA 的這個網(wǎng)站頁面(https://developer.nvidia.com/cuda-gpus),點開你電腦顯卡對應的系列诸迟,查看顯卡中GPU的計算能力(Compute Capability)
geforce-gtx-1080 算力6.1
cuDNN解壓出來三個文件夾+一個文件壹粟,將各個文件夾下面的文件分別放到CUDA對應的同名文件夾下。
我的CUDA地址是默認地址C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
完成之后把以下四個路徑加入到環(huán)境變量中:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
2. 安裝Anaconda
來源:
- Anaconda官網(wǎng)下載速度特別慢
- 清華開源鏡像網(wǎng)站(https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/)
安裝時勾選“添加Anaconda環(huán)境變量 ”選項
3. 安裝TensorFlow GPU
沒有配置環(huán)境呻此,直接安裝的tf
×【安裝了但是不知道為什么不能用】在Anaconda prompt里使用conda安裝(也可在terminal里用pip安裝)轮纫,順便裝了keras
#conda install tensorflow
conda install tensorflow-gpu
conda install keras
pip3 install tensorflow-gpu==2.0.0
#或
pip3 install --upgrade --ignore-installed tensorflow-gpu==2.0.0
出現(xiàn)紅色一片,報錯超時焚鲜,延長超時時間:
pip3 install --upgrade --ignore-installed tensorflow-gpu==2.0.0 --default-timeout=100
#或
pip3 install --upgrade --ignore-installed tensorflow-gpu --default-timeout=100
√【重新在cmd terminal里安裝的】
pip install tensorflow-gpu
pip install keras
4. 測試|TF GPU
進入python界面或打開jupyter notebook測試
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
或
import tensorflow as tf
tf.__version__
5. 配置 jupyter notebook 遠程連接
① 生成 notebook 配置文件
默認情況下配置文件 ~/.jupyter/jupyter_notebook_config.py (所在文件夾:C:\Users\user.jupyter)不存在掌唾,要自行創(chuàng)建,使用以下命令生成配置文件:
jupyter notebook --generate-config
root 用戶執(zhí)行時需要加上 --allow-root 選項忿磅。
jupyter notebook --generate-config --allow-config
② 生成密碼
自動生成密碼:從 jupyter notebook 5.0 版本開始糯彬,提供了一個命令來設置密碼:jupyter notebook password,生成的密碼存儲在 jupyter_notebook_config.json葱她。
在jupyter_notebook_config.py 里相應行添加你的密碼撩扒。
c.NotebookApp.password =你的密碼
③ 修改配置文件
在 jupyter_notebook_config.py 中找到下面的行,取消注釋并修改吨些。
c.NotebookApp.ip='*'
c.NotebookApp.password = u'sha:ce...剛才復制的那個密文'
c.NotebookApp.open_browser = False
c.NotebookApp.port =8888 #可自行指定一個端口, 訪問時使用該端
以上設置完以后就可以在服務器上啟動 jupyter notebook搓谆,jupyter notebook, root 用戶使用 jupyter notebook --allow-root炒辉。
打開 IP:指定的端口, 輸入密碼就可以訪問了。