win10+英伟达mx150安装cuda,cudnn,GPU版本pytorch(荣耀magicbook2018)

最近打算给笔记本装一个gpu版本的pytorch,装了有半天才搞完,中间也出现了不少问题看了不少博客,现在总结一下出现的问题以及解决办法,希望能给遇到同样问题的提供一些帮助。本人水平有限,有问题还请指正。

我这里装的是pytorch1.6.2版本,CUDA对应9.2,建议各位装pytorch之前,首先确定你确实需要什么版本的pytorch,在根据你设备的具体情况酌情安装。

 

  1. 安装cuda

    如果想使用gpu版本的pytorch,首先需要安装一个自己显卡支持的cuda版本
    我的电脑显卡是mx150,cpu是i7-8550U,操作系统是win10,所以我这里选择的是cuda9.2版本win10+英伟达mx150安装cuda,cudnn,GPU版本pytorch(荣耀magicbook2018)_第1张图片https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html         可以查看安装cuda所需要的显卡驱动版本,查看电脑的显卡驱动版本,右键选择NVIDIA 控制面板->帮助->系统信息win10+英伟达mx150安装cuda,cudnn,GPU版本pytorch(荣耀magicbook2018)_第2张图片

  2. 显卡驱动版本只要大于cuda对应的最低版本就行,如果显卡驱动版本过低,可以去英伟达的官网下载对应的驱动

你可能感兴趣的:(pytorch,深度学习,python)