ESM2模型设置##加载本地的tokenizer和模型定义一个新的类，它继承自PyTorch的Dataset类。这个类用于创建一个自定义的数据集，可以与DataLoader一起使用来加载数据。__init__...

浏览 52 次 标签： 计算生物——Code_Pytorch框架——蛋白靶点小分子药物对接亲和力预测_CPU（07.11-07.22）   

我这里有一台从算法组借用的GPU服务器，里边是两张。使用命令查看当前的nvidia-smi。Ubuntu18服务器更新升级CUDA版本pyenvnvidiaubuntu1804原11.2升级到PyT...

浏览 59 次 标签： Ubuntu18 服务器 更新升级CUDA版本 pyenv nvidia ubuntu1804 原11.2升级到PyTorch要求12.1 全过程详细记录 apt update   

大家好，我是默语，擅长全栈开发、运维和人工智能技术。在PyTorch模型的推理阶段，RuntimeError是常见的问题之一。这类错误通常会导致模型推理延迟，严重影响模型的实时性能和用户体验。本篇博客将深入探讨R...

浏览 97 次 标签： 推理延迟：解决PyTorch模型Inference阶段的RuntimeError ⏳⚡   

带你快速上手语义分割经典项目...

浏览 92 次 标签： Ubuntu系统DeeplabV3Plus实战（PyTorch版本）   

当你遇到在Anaconda环境中通过命令行可以成功导入PyTorch模块，但在JupyterNotebook中却提示找不到模块的情况时，这通常是因为JupyterNotebook。打开你的Notebook，点击...

浏览 63 次 标签： Import torch 报错    称找不到模块的解决办法   

本文用于总结前几天在配置python环境的过程中碰到的种种问题，以及对自己看到的数不清的博客做一个总结归纳，将自己坎坷的安装史留次记录，希望可以给到其他人一点帮助。同时，这两天在用树莓派的时候也避不开配置pyth...

浏览 55 次 标签： Anaconda和GPU版pytorch安装过程一条龙整理以及报错总结   

torch.nn.RNN(）相关的参数设置...

浏览 96 次 标签： torch.nn.RNN(）相关的参数设置   

重新安装torch，输入命令：pipinstalltorch==2.0.0，重新安装就可以了。4.以下是torch以及torchvision和python的版本对应图。2.报错原因：是因为torch和python版本不对应。_导入to...

浏览 66 次 标签： 报错 Error loading “C:\Users\Thinkpad\AppData\Roaming\Python\Python311\site-packages\torch\li    导入torch时   

今天要介绍的这段代码是一个使用PyTorch框架实现的卷积神经网络（CNN）模型，用于对MNIST数据集进行分类的示例。MNIST数据集是手写数字识别领域的一个标准数据集，包含0到9的灰度图像。导入必要的库：导入...

浏览 80 次 标签： PyTorch--卷积神经网络（CNN）模型实现手写数字识别   

如果安装cuda11.8版本的，把最后的cu121改成cu118即可。目前支持最高cuda12.2。因为网上pipinstalltorch等全部默认安装cpu版本，推理速度慢。所有安装cuda版本的包会更...

浏览 92 次 标签： torch、torchvision、torchaudio安装对应cuda版本（cuda 12.1情况下）