【自用】笔记本安装pytorch全流程

记录一下自己安装pytorch的流程。笔记本电脑配置不高,NVIDIA MX250

一共有三个步骤:

  1. 确定电脑有支持的GPU,安装Cuda
  2. 安装cuDNN
  3. 安装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,
重复上面的安装步骤,提示成功并测试。
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值