我的硬件环境是: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下装,具体请参考搜索引擎给出的答案。?