1.创建环境
conda create -n tf1.9 python=3.6
激活自己的环境
source activate tf1.9
2.用pytorch安装cuda
找好pytorch与cuda的对应关系
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
3.再安装想要安的tensorflow
!tensorflow最好用pip安装,比conda好用
pip install tensorflow-gpu==1.9
安装tensorflow-gpu1.9对应keras
pip install keras==2.2.0
4.检查cuda是否安装成功
nvcc -V
5.安装各种模块
pip install matplotlib
pip install opencv-python
pip install imageio
pip install scikit-image
sudo apt-get install git
pip install git+https://github.com/JiahuiYu/neuralgym
pip install scipy==1.2.1