HIPPT·手把手教你【深度学习】项目实战

原创

已于 2024-03-21 16:59:55 修改 · 1.2k 阅读

17 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #人工智能 #机器学习 #python #docker #计算机视觉 #目标检测

于 2024-03-19 20:48:57 首次发布

这篇博客介绍了如何在深度学习项目中使用PyTorch和Einops库。首先，详细讲解了PyTorch的安装过程，包括选择适合的CUDA版本和确保在正确环境中安装。接着，指导如何启动和配置Jupyter Notebook，以及如何在新环境中检查和安装缺失的依赖库，如einops、cv2等。最后，提到了在使用HIPT_4K应用程序接口时遇到的问题，包括一个未定义的变量，以及解决这个问题的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

上一篇推文我们学会了如何在服务器内克隆仓库，如何初步运行代码，在这一篇推文我们将进一步学习如何处理复杂的项目。

这段代码涉及了PyTorch库、einops库以及自定义的函数调用。

import torch
from einops import rearrange, repeat
from HIPT_4K.hipt_model_utils import get_vit256, get_vit4k

import torch: 这行代码导入了PyTorch库，PyTorch是一个用于深度学习的开源机器学习库，提供了张量运算和神经网络构建等功能。
from einops import rearrange, repeat: 这行代码从einops库中导入了rearrange和repeat函数。einops是一个**用于重塑和操作张量**的库，它提供了简单而强大的工具来处理张量的形状。
from HIPT_4K.hipt_model_utils import get_vit256, get_vit4k: 这行代码从自定义的"HIPT_4K"模块中导入了get_vit256和get_vit4k函数。

总体来说，这段代码的作用是导入PyTorch库、einops库中的函数，并从自定义模块中导入特定的函数以便后续使用。

⚠️：为了不增加你的难度，请你现在不要打开jupyter！！！看我的界面就行

目前，我们遇到了第一个问题，未安装Pytorch。

一、Pytorch的安装

注意⚠️

如果此时你是重新开始或者第一次按照这个教程操作，请一定要先保证自己已经进入之前创造的虚拟环境了。

cd learn
cd HIPT
conda activate HIPT

查看CUDA版本

nvidia-smi

进入pytorch下载页面

挑选适合自己的版本

这里注意，我第一次尝试时，挂着梯子，但是报错了，所以最好在导入包时，把梯子先关掉。

安装成功，检查一下。

conda list

二、Jupyter操作

启动jupyter

jupyter notebook --no-browser

打开终端

输入如下代码

ssh -N -L localhost:8888:localhost:8888 username@ip地址

浏览器配置

在上一步正确输入密码之后，打开你的浏览器，在网址栏输入localhost:8888后，会提示你输入之前设置的密码，验证成功

最低0.47元/天解锁文章

200万优质内容无限畅学