本文主要参考的是知乎上的一篇《PyTorch在64位Windows下的Conda包》:https://zhuanlan.zhihu.com/p/26871672
该安装十分简单,主要是用了楼上作者提供的PyTorch的windows conda包,该包中内置CUDA 8.0,cuDNN v6(cuDNN v5的用户也可以使用)。
配置要求
- Anaconda3(with Python 3.6)
- Windows7及以上
本文主要参考的是知乎上的一篇《PyTorch在64位Windows下的Conda包》:https://zhuanlan.zhihu.com/p/26871672
该安装十分简单,主要是用了楼上作者提供的PyTorch的windows conda包,该包中内置CUDA 8.0,cuDNN v6(cuDNN v5的用户也可以使用)。