环境
硬件环境
我的硬件环境是:Intel I7 8750H, GTX 1060
系统环境
anaconda 4.5.11
Ubuntu18.04
软件环境
名称 |
版本 |
python |
3.6.7 |
cuda |
9.0 |
cudnn |
7.4 |
Tensorflow-gpu |
1.8.0 |
Keras |
2.1.5 |
nltk |
3.2.5 |
Pillow |
5.0.0 |
OpenCV |
3.4.1 |
PyTorch |
0.4.1 |
安装过程在此省略
PS.用conda装PyTorch0.4.1时有问题,因为现在(2019年10月17日)PyTorch的版本已经是1.X了,直接是装不上的。要装0.4.1-gpu版本,需要在pip下装,具体请参考搜索引擎给出的答案。?