深度学习系列错误笔记(一)之print(‘gpu‘,torch.cuda.is_available())输出gpu False

5 篇文章 0 订阅
2 篇文章 0 订阅

0.问题出现背景

ps: win10 64位环境。cuda已经安装,显卡驱动版本472对应torch1.8.0版本对应cuda10.2版本都没问题。

因为硬盘坏了,所以从头开始装按照深度学习系列(一)从零相关软件安装篇(笔记)_lry320的博客-CSDN博客

 安装最新的发现,最后

>>>print('gpu',torch.cuda.is_available())

gpu False

 后来发现可能是因为我的Anaconda改了清华源,虽然用的官网Previous PyTorch Versions | PyTorch

的命令下载torch1.8带GPU的命令,但是却只能下载cpu的版本。

1.解决方法

a.去官网Start Locally | PyTorch选择pip使用网站下载

pip3 install torch==1.8.2+cu102 torchvision==0.9.2+cu102 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

 运行结果如下:

>>> import torch
>>> print(torch.__version__)
1.8.2+cu102
>>> print('gpu',torch.cuda.is_available())
gpu True

b.离线环境下通过Anaconda安装相关包

手动去Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

找到自己对应的版本下载。 此处我选择了最新的pytorch-1.9.1-py3.9_cuda10.2_cudnn7_0.tar.bz2

 版本对应表格https://github.com/pytorch/vision#installation,最新下载的Anaconda自带的Python3.9版本。注意安装的torchvison版本即可。

下载好压缩包后,用cmd进入解压的文件夹中(直接在当前文件夹位置地址栏输入%comspec%,进入该目录的命令提示符),运行命令

conda install pytorch-1.9.1-py3.9_cuda10.2_cudnn7_0.tar.bz2

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值