安装pytorch时,torch.cuda.is_available()返回false

记录一下自己安装pytorch后显示torch.cuda.is_available()的解决过程。

1. 查看gpu的型号:

打开任务管理器,选择性能,选择GPU0(或GPU),会在界面显示gpu的型号

2. 查看cuda和cudnn的安装情况

在文件夹C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA下查看安装的cuda的版本,例如该文件夹下有v10.0,则代表安装的是v10.0版本的cuda.

在 cmd 中输入 nvidia-smi,同时这个指令也可以查看 GPU 的一些信息,如果出现如下界面就说明 CUDA 安装成功了

3. 找到和cuda版本以及python版本匹配的离线安装包进行安装。在线安装容易安装成cpu版本的,或者连不上包管理服务器。 

离线包下载地址

https://download.pytorch.org/whl/torch_stable.html

例如本人电脑上安装的是python3.7, cuda版本是10.0,则下载:cu100/torch-1.2.0-cp37-cp37m-win_amd64.whl cu100/torchvision-0.4.0-cp37-cp37m-win_amd64.whl 这两个文件

  • 7
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值