PyTorch在win10-64位系统下搭建的全过程

https://blog.csdn.net/sinat_23619409/article/details/84202651
首先,要安装cuda和cudnn,上面的链接写的非常全面。
https://developer.nvidia.com/cuda-toolkit-archive
然后pytorch,要确保显卡支持的cuda、安装的cuda、pytorch版本都一致,我这里版本全是9.2
显卡支持的 cuda在控制面板-nvidia控制面板-帮助-系统信息-组件查看。
最后用conda下的python.exe测试:
import torch
torch.cuda.is_available()
PyTorch在win10-64位系统下搭建的全过程_第1张图片

你可能感兴趣的:(pytorch)