记录一下自己安装pytorch的流程。笔记本电脑配置不高,NVIDIA MX250
一共有三个步骤:
- 确定电脑有支持的GPU,安装Cuda
- 安装cuDNN
- 安装PyTorch
1. 安装CudaToolkit
(1)首先,查看电脑N卡版本
第一种方式,NVIDIA控制面板-帮助-系统信息,这里直接可以看到适配的cuda版本
第二种方式,打开任务管理器查看
然后去官网查看对应算力匹配的cuda版本
(2)安装N卡对应的Cuda Toolkit
官网:https://developer.nvidia.com/cuda-toolkit-archive
安装软件的过程参考[这位作者]
2.安装cuda对应的cuDNN(可选,加速用)
使用如下命令查看conda支持的cudnn版本:conda search cudnn --info
这里面会列出所有CudaToolkit版本对应的cudnn的下载链接URL,复制到浏览器下载即可。
下载完成后使用指令进行本地安装
conda install --use-local D:\download\cudnn-7.6.5-cuda10.0_0.conda
3. 安装PyTorch
下面这个网站查询cuda对应安装哪个torch版本:
https://pytorch.org/get-started/previous-versions/
到轮子网站(https://download.pytorch.org/whl/torch_stable.html)下载torch
在anaconda prompt里面,激活相应的环境,准备安装。
首先切换到刚刚下载whl的路径的D盘
然后发现报错,提示下载安装的包不是当前平台所支持的。
解决办法:
pip debug --verbose
查看哪些才是支持的版本,结果如图
发现原来是python的版本3.7有点高了,于是回去轮子网站重新下载了torch-1.2.0-cp36-cp36m-win_amd64.whl,
重复上面的安装步骤,提示成功并测试。