PyTorch GPU安装指南 (Ubuntu 16.04 anaconda cuda8.0 cuDNN6.0)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/chenhaifeng2016/article/details/78874840

1. python

2. 显卡驱动

3. Nvidia CUDA

4. nvidia CuDNN

以上安装过程和Tensorflow相同,请参照 

http://blog.csdn.net/chenhaifeng2016/article/details/78874883


5. PyTorch




6. 开发工具PyCharm

安装过程略


7.测试




猜你喜欢

转载自blog.csdn.net/chenhaifeng2016/article/details/78874840