https://blog.csdn.net/sinat_23619409/article/details/84202651
首先,要安装cuda和cudnn,上面的链接写的非常全面。
https://developer.nvidia.com/cuda-toolkit-archive
然后pytorch,要用pip命令安装,要确保显卡支持的cuda、安装的cuda、pytorch版本都一致,我这里版本全是9.2。
如果更新的话,torch直接使用pip命令安装最新版就行,conda会自动卸载老版本,CUDA需要手动卸载老版本。
显卡支持的 cuda在:控制面板-nvidia控制面板-帮助-系统信息-组件查看。
最后用conda下的python.exe测试:
import torch
torch.cuda.is_available()
PyTorch在anaconda+win10-64位系统下搭建的全过程
最新推荐文章于 2024-07-25 14:35:42 发布