深度学习——利用GPU进行训练数据——安装错误的修正

当torch.cuda.is_available()返回false时,需检查PyTorch安装是否为GPU版本。可通过conda或pip卸载并重新安装,确保从官网下载匹配CUDA版本的PyTorch。安装时,应选择低于NVIDIACUDA当前版本的兼容版本,并根据Python版本选择正确的whl文件。安装后,验证torch.cuda.is_available()应返回True。
摘要由CSDN通过智能技术生成

安装pytorch时,torch.cuda.is_available()返回false。上网查阅资料,输入​​conda list​​查看从官网下载的pytorch是CPU版本还是GPU版本。
在这里插入图片描述显然,我安装的是cup版本。
在这里插入图片描述

一、卸载pytorch

使用conda卸载Pytorch
1.conda uninstall pytorch
2.conda uninstall libtorch
使用pip卸载Pytorch
pip uninstall torch

如果当初是用conda install 命令安装的pytorch,则用第一种方法,如果是用pip install命令安装的,则用第二种方法。或者两种方法都试一下

去官网使用PIP安装方式安装新的pytorch,记住千万,千万不要用conda的安装方式,因为它下载的一直都是CPU版本的。(我的教训)
若自己下载的确实为GPU版本的,但是仍旧出现torch.cuda.is_available()返回false的情况,则可能是版本不匹配的问题。

二、安装pytorch(我已经安装了cuda,版本11.3)

cuda安装步骤,我觉得这个不错。
查看自己的NAVIDA的CUDA版本,win+R,cmd命令行输入nvidia-smi,一定要选择比自己版本低的CUDA。比如,我的版本是11.6,那么就不能选择11.7,我选择的是11.3。
在这里插入图片描述

不使用命令行下载pytorch

https://download.pytorch.org/whl/torch_stable.html
发现第一个就是你的cuda版本 第二个是你要下的torch(或torchvision)版本号,第三个是对应的你的py版本号,然后是选择linux或者windows。

我的cuda是11.3,py是3.8,torch版本根据自己需求进行选择。

在这里插入图片描述在这里插入图片描述

下载torchversion

在这里插入图片描述

打开
在这里插入图片描述
当用上述方法,安装完两个库之后。
cmd 操作:cd找到对应安装位置。
然后pip install 名字.whl;这两个文件就都被安装了
在这里插入图片描述
中间可能会因为顺序原因,报错
eg.
ERROR: torchvision 0.5.0 has requirement torch==1.4.0, but you’ll have torch 1.5.0+cu101 which is incompatible.
问题不大,两个包按顺序各按几次。
在这里插入图片描述最后进行测试:
先输入python

import torch
torch.cuda.is_available()

输出True说明安好了。
在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值