1.安装python和python-pip
1 | $ sudo apt install python3 |
2.安装tensorflow-gpu版本
1 | $ pip install tensorflow-gpu |
3.安装IDE,修改配置文件,开启远程
我这里使用的是jupyter-notebook
1 | $ sudo python3 -m pip install jupyter |
修改配置文件,密码为 a123456
此处密码生成规则使用 notebook.auth 就可以
1 | c.NotebookApp.ip = '*' |
然使用 jupyter notebook
即可启动,访问 http://ip:9999/
4.在jupyter中测试tensorflow
1 | import tensorflow as tf |
我这里会报错,因为我安装的是最新的GPU版本需要CUDA9.0支持
1 | ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory |
从 Nvidia官网 下载安装最新cuda
1 | $ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb |
5.实现线回归
1 | #实现线性回归 |