U-net的環境配置

Htk_Dn發表於2020-11-22

U-net的環境配置

一、跑U-net程式碼需安裝下列軟體和模組:

1、軟體:
CUDA9.0
cuDNN7.0
Anaconda

2、模組:
TensorFlow-gpu 1.9
keras 2.2.0
numpy 1.16(安裝最新版的話可能會不相容,故選了1.16版本)
scikit-image
h5py

二、安裝過程:

1、在Anaconda Prompt建立虛擬環境(本程式碼最高相容到python3.5):

conda create -n Unet python=3.5

2、啟用環境:

conda activate Unet

3、依次輸入以下指令:

pip install tensorflow-gpu==1.9
pip install keras==2.2.0
pip install scikit-image
pip install numpy==1.16
pip install h5py