将GPU版本的torch和torchvision下载至本地进行安装,解决torch.cuda.is_available()为False的情况,本文主要解决CPU版本的torch问题。

这是困扰我很久的问题,今天花了一点时间去研究一下为什么我的cuda和cudann安装完成了,却不能使用GPU,原因是我的torch和torchvision是cpu版本的,必须下载GPU版本的才能使torch.cuda.is_available()为True,本文使用将文件下载到文件夹后进行安装。

1、打开Anaconda Prompt(Anaconda3)

 2、打开后是以base开头的对话框

 使用conda activate opencv     此处opencv是我自己创建的环境名称,此处更换为你自己的环境名

进入你自己需要安装torch的位置,输入python,进入python中

import torch
torch.cuda.is_available()

因为我的电脑解决了这个问题,所以是True,如果您的电脑是False,查看是否是CPU版本的torch.

 输入torch.__version__   如果返回cpu字样ÿ

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值