自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(4)
  • 收藏
  • 关注

原创 docker容器中torch.cuda.is_available()为false

网上搜了大部分都说是重启解决,再者就是pytorch的版本和cuda版本不兼容,但我的image直接下载的就是docker hub中的pytorch/pytorch,不应该会出现版本不兼容的问题。接下来再去寻找其他的方式,最后试下来是要重装显卡驱动,回退版本(也有说只需重装就行,无需回退版本)现象:运行了torch.cuda.is_available()后显示。

2024-06-09 18:09:32 254

原创 Docker实例化容器后,报错:WARNING: The NVIDIA Driver was not detected. GPU functionality will not be availabl

Docker实例化容器后,报错:WARNING: The NVIDIA Driver was not detected. GPU functionality will not be available。

2024-06-01 01:24:35 445 2

原创 用torch编写机器学习代码,如何默认使用GPU

作用:首先检查CUDA是否可用,如果可用,则将默认张量类型设置为CUDA的浮点张量。

2024-02-26 00:06:45 481 1

原创 torch.cuda.device_count()结果为0,检测不到显卡

检查自己的python版本、cuda版本、torch和 torchvision 版本是否兼容。例:cuda12.1,python 3.9,torchversion 0.16.0,win。例:cuda12.1,python 3.9,torch2.1.0,win。寻找对应的torch和 torchvision 版本下载地址。cd到下载目录下pip安装。安装前先卸载原有版本。

2024-02-15 16:32:59 1716

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除