4070显卡配置ubuntu深度学习环境

2023-12-19 21:01:25

本文主要记录一下自己深度学习环境的配置,具体的安装教程有很多,这里就不赘述了。

1.显卡型号

华硕 ATS GeForce RTX4070-o12G-GAMING

2.系统和驱动型号

ubuntu 20.04? 显卡驱动版本:NVIDIA-Linux-x86_64-535.129.03.run

3.cuda和cudnn版本

cuda :

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0

cudnn:

cudnn-linux-x86_64-8.9.5.30_cuda11-archive

4.pytorch版本

通过anaconda配置的环境,在官网找的命令安装的,版本是2.1.0+cu118

在ubuntu中cuda版本要完全对应才能用,比如我的全是11.8版本。我装的双系统,Windows系统中没有单独安装cuda和cudnn,直接通过anaconda配置的pytorch,版本是torch1.12+cuda11.3,也能正常使用。

ubuntu中配置的深度学习环境跑yolov8是可以正常训练的。

文章来源:https://blog.csdn.net/m0_51530640/article/details/135082428
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。