win10+英伟达mx150安装cuda,cudnn,GPU版本pytorch | 您所在的位置:网站首页 › m1支持cuda吗 › win10+英伟达mx150安装cuda,cudnn,GPU版本pytorch |
最近打算给笔记本装一个gpu版本的pytorch,装了有半天才搞完,中间也出现了不少问题看了不少博客,现在总结一下出现的问题以及解决办法,希望能给遇到同样问题的提供一些帮助。本人水平有限,有问题还请指正。 1.安装cuda如果想使用gpu版本的pytorch,首先需要安装一个自己显卡支持的cuda版本 我的电脑显卡是mx150,cpu是i5-8250u,操作系统是win10系统 此处选择的是cuda9.2版本 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html可以查看安装cuda所需要的显卡驱动版本,查看电脑的显卡驱动版本,右键选择NVIDIA 控制面板->帮助->系统信息 cuDNN Archive | NVIDIA Developer 需要先注册才能下载,选择对应cuda的版本,这里选择cuda9.2的对应版本 接下来测试cudnn安装是否成功 在extras/demo_suite/目录下右键在windows终端中打开,然后./bandwidthTest.exe运行 如果Result=PASS则代表cudnn安装成功 由于我的电脑已经安装过anaconda就不再重复此操作 下载地址为https://www.anaconda.com/ 安装时注意勾选将anaconda加入到环境变量中 打开Anaconda Promt 通过conda创建一个名为pytorch的虚拟环境 conda create -n pytorch python=3.7可以使用如下命令查看已经创建好的虚拟环境 conda info --envs
运行之后命令行前面的(base)会变成(pytorch)表示已经切换到pytorch环境中 在清华源镜像中下载pytorch,torchvision,cuda92 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ 安装完成之后如图所示
|
CopyRight 2018-2019 实验室设备网 版权所有 |