博客
关于我
linux下安装GPU 版本 torch torchvision等最方便的方式
阅读量:789 次
发布时间:2023-02-02

本文共 478 字,大约阅读时间需要 1 分钟。

安装YOLOV5的步骤其实非常简单快捷。只需按照项目文档中的要求,只需在本地项目目录下找到requirements.txt文件,将其内容复制到你的项目目录下新建或更改名为requirements.txt,然后在终端中运行pip install -r requirements.txt即可。

更换安装源提升速度

为了加快安装速度,可以选择使用更靠近的镜像源。例如,在安装PyTorch等依赖时选择以下命令:

conda install pytorch torchvision torchaudio cudatoolkit -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

验证代码示例

安装完成后,验证是否正确安装可以运行以下代码:

a = torch.Tensor([1,2])a = a.cuda("cuda:0")print(a)

输出结果应为:

tensor([1. , 2.])

以上就是完整的安装和验证流程,可以根据实际需求灵活配置安装源和其他参数。

转载地址:http://idzfk.baihongyu.com/

你可能感兴趣的文章