超详细||深度学习环境搭建记录cuda+anaconda+pytorch+pycharm

这篇具有很好参考价值的文章主要介绍了超详细||深度学习环境搭建记录cuda+anaconda+pytorch+pycharm。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

本文用来记录windows系统上深度学习的环境搭建,目录如下

一、安装显卡驱动

首先为装有NVIDIA gpu的电脑安装显卡驱动,如果安装过了,或者想使用cpu的,可以跳过这一步。(其实这一步可以跳过,因为显卡驱动好想和深度学习环境没什么关系,保险起见还是安装上吧)

1. 去官网下载对应的显卡驱动:官方驱动 | NVIDIA     2023 深度学习环境搭建,深度学习,人工智能

2. 完成下载,选择文件开始安装,直接解压在默认地址

2023 深度学习环境搭建,深度学习,人工智能

3. 选择自定义安装选项,执行清洁安装(按情况选择)

2023 深度学习环境搭建,深度学习,人工智能

4. 一直点下一步即可。

二、安装Visual Studio

可以跳过,但是很多深度学习环境需要用到,建议安装

1. 官网下载 Visual Studio Tools - 免费安装 Windows、Mac、Linux (microsoft.com)

2023 深度学习环境搭建,深度学习,人工智能

2. 选择python开发和c++桌面开发,开始安装

2023 深度学习环境搭建,深度学习,人工智能

三、cuda+cudnn安装

1. 打开cmd,输入nvidia-smi查看GPU的CUDA版本,可以看到CUDA版本12.1,意味着只能安装小于12.1的CUDA版本。

2023 深度学习环境搭建,深度学习,人工智能


CUDA Toolkit Archive | NVIDIA Developer2. 去网站下载适合的CUDA版本:CUDA Toolkit Archive | NVIDIA Developer

要根据两个条件选择:

  • 1. CUDA版本要小于上面的版本信息
  • 2. 先去pytorch官网Start Locally | PyTorch,看一眼自己需要的pytorch版本对应的CUDA版本

例如:pytorch1.12.0只支持CUDA10.2,11.3,11.6

2023 深度学习环境搭建,深度学习,人工智能

3. 然后去官网下载:CUDA Toolkit Archive | NVIDIA Developer,以11.6版本举例,选好系统参数,点击下载,

2023 深度学习环境搭建,深度学习,人工智能

4. 下载完成,直接默认解压位置

2023 深度学习环境搭建,深度学习,人工智能

5. 默认勾选即可,点下一步

2023 深度学习环境搭建,深度学习,人工智能
 

6. 安装cudnn

去官网下载CUDA Deep Neural Network (cuDNN) | NVIDIA Developer,需要注册一个NVIDIA账号,然后选择CUDA对应版本的cudnn即可。

2023 深度学习环境搭建,深度学习,人工智能

7. 下载cudnn后直接将其解开压缩包然后需要将解压后的bin,include,lib文件夹复制粘贴到cuda安装时的默认路径文件夹下

2023 深度学习环境搭建,深度学习,人工智能

 8. 检查环境变量

2023 深度学习环境搭建,深度学习,人工智能

2023 深度学习环境搭建,深度学习,人工智能

9. 在cmd输入nvcc -V,显示CUDA版本即安装成功

2023 深度学习环境搭建,深度学习,人工智能

安装多个CUDA环境:

如果要安装多个CUDA环境的话,和上述步骤基本相同,其中在安装时只需要选中cuda即可

2023 深度学习环境搭建,深度学习,人工智能

如果要在各个CUDA间进行转换,只需要修改环境变量即可。

例如:把CUDA11.6改为CUDA11.8,只需要把CUDA_PATH改为v11.8,并把Path中v11.6的路径上移到v11.8之前,然后重启电脑即可

2023 深度学习环境搭建,深度学习,人工智能

2023 深度学习环境搭建,深度学习,人工智能

四、anaconda安装

1. 去官网下载安装包:Free Download | Anaconda

2. 下载完成,进行安装,建议安装在非系统盘,以后会创建比较多的环境的话,剩下的默认安装即可

2023 深度学习环境搭建,深度学习,人工智能

五、pycharm安装

1. 官网下载安装包:PyCharm: the Python IDE for Professional Developers by JetBrains,可申请教育账号免费下载专业版,如果没有,下载社区版即可

2. 自行安装即可

六、pytorch安装,及在pycharm中选择conda环境

1.首先创建一个conda环境,打开Anaconda prompt,输入:conda create --name yourname python=3.10 -y

删除环境输入: conda remove -n env_name --all

2023 深度学习环境搭建,深度学习,人工智能

2. 输入conda activate yourname 激活环境

2023 深度学习环境搭建,深度学习,人工智能

2. 去官网找自己需要的pytorch版本:Start Locally | PyTorch 。选择电脑参数,并复制pip install命令,也可以去找之前的版本。

2023 深度学习环境搭建,深度学习,人工智能

3. 将复制的pip install命令,粘贴在Anaconda prompt中,等待安装即可

2023 深度学习环境搭建,深度学习,人工智能

4.打开anaconda prompt命令行测试代码

python 
import torch 
print(torch.__version__)
print(torch.cuda.is_available())

5.打开pycharm,选择添加新的解释器→添加本地解释器→conda环境→/yourname/python.exe (这里不同版本的pycharm可能有点不一样,只要能识别就行)

x2023 深度学习环境搭建,深度学习,人工智能

2023 深度学习环境搭建,深度学习,人工智能

至此,环境搭建结束。。。。

安装了yolov8的,可以试着训练一下yolov8文章来源地址https://www.toymoban.com/news/detail-813031.html

from ultralytics import YOLO
 
if __name__ == '__main__':
    # Load a model
    model = YOLO(r'\ultralytics\detection\yolov8n\yolov8n.yaml')  # 不使用预训练权重训练
    # model = YOLO(r'yolov8p.yaml').load("yolov8n.pt")  # 使用预训练权重训练
    # Trainparameters ----------------------------------------------------------------------------------------------
    model.train(
        data=r'\ultralytics\detection\dataset\appledata.yaml',
        epochs= 30 , # (int) number of epochs to train for
        patience= 50 , # (int) epochs to wait for no observable improvement for early stopping of training
        batch= 8 , # (int) number of images per batch (-1 for AutoBatch)
        imgsz= 320 , # (int) size of input images as integer or w,h
        save= True , # (bool) save train checkpoints and predict results
        save_period= -1, # (int) Save checkpoint every x epochs (disabled if < 1)
        cache= False , # (bool) True/ram, disk or False. Use cache for data loading
        device= 0 , # (int | str | list, optional) device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu
        workers= 16 , # (int) number of worker threads for data loading (per RANK if DDP)
        project= 'result', # (str, optional) project name
        name= 'yolov8n' ,# (str, optional) experiment name, results saved to 'project/name' directory
        exist_ok= False , # (bool) whether to overwrite existing experiment
        pretrained= False , # (bool | str) whether to use a pretrained model (bool) or a model to load weights from (str)
        optimizer= 'SGD',  # (str) optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto]
        verbose= True ,# (bool) whether to print verbose output
        seed= 0 , # (int) random seed for reproducibility
        deterministic= True , # (bool) whether to enable deterministic mode
        single_cls= True , # (bool) train multi-class data as single-class
        rect= False  ,# (bool) rectangular training if mode='train' or rectangular validation if mode='val'
        cos_lr= False , # (bool) use cosine learning rate scheduler
        close_mosaic= 0,  # (int) disable mosaic augmentation for final epochs
        resume= False , # (bool) resume training from last checkpoint
        amp= False,  # (bool) Automatic Mixed Precision (AMP) training, choices=[True, False], True runs AMP check
        fraction= 1.0 , # (float) dataset fraction to train on (default is 1.0, all images in train set)
        profile= False,  # (bool) profile ONNX and TensorRT speeds during training for loggers
        # Segmentation
        overlap_mask= True , # (bool) masks should overlap during training (segment train only)
        mask_ratio= 4,  # (int) mask downsample ratio (segment train only)
        # Classification
        dropout= 0.0,  # (float) use dropout regularization (classify train only)
        # Hyperparameters ----------------------------------------------------------------------------------------------
        lr0=0.01,  # (float) initial learning rate (i.e. SGD=1E-2, Adam=1E-3)
        lrf=0.01,  # (float) final learning rate (lr0 * lrf)
        momentum=0.937,  # (float) SGD momentum/Adam beta1
        weight_decay=0.0005,  # (float) optimizer weight decay 5e-4
        warmup_epochs=3.0,  # (float) warmup epochs (fractions ok)
        warmup_momentum=0.8,  # (float) warmup initial momentum
        warmup_bias_lr=0.1,  # (float) warmup initial bias lr
        box=7.5,  # (float) box loss gain
        cls=0.5,  # (float) cls loss gain (scale with pixels)
        dfl=1.5,  # (float) dfl loss gain
        pose=12.0,  # (float) pose loss gain
        kobj=1.0,  # (float) keypoint obj loss gain
        label_smoothing=0.0,  # (float) label smoothing (fraction)
        nbs=64,  # (int) nominal batch size
        hsv_h=0.015,  # (float) image HSV-Hue augmentation (fraction)
        hsv_s=0.7,  # (float) image HSV-Saturation augmentation (fraction)
        hsv_v=0.4,  # (float) image HSV-Value augmentation (fraction)
        degrees=0.0,  # (float) image rotation (+/- deg)
        translate=0.1,  # (float) image translation (+/- fraction)
        scale=0.5,  # (float) image scale (+/- gain)
        shear=0.0,  # (float) image shear (+/- deg)
        perspective=0.0,  # (float) image perspective (+/- fraction), range 0-0.001
        flipud=0.0,  # (float) image flip up-down (probability)
        fliplr=0.5,  # (float) image flip left-right (probability)
        mosaic=1.0,  # (float) image mosaic (probability)
        mixup=0.0,  # (float) image mixup (probability)
        copy_paste=0.0,  # (float) segment copy-paste (probability)
                )
 

到了这里,关于超详细||深度学习环境搭建记录cuda+anaconda+pytorch+pycharm的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 从零开始配置深度学习环境:CUDA+Anaconda+Pytorch+TensorFlow

    本文适用于电脑有GPU(显卡)的同学,没有的话直接安装cpu版是简单的。CUDA是系统调用GPU所必须的,所以教程从安装CUDA开始。 可以配合视频教程食用:https://www.bilibili.com/video/BV12m4y1m7pq/?vd_source=06e4e8652ea90d79dadb7a59ff8acd36 CUDA是加速深度学习计算的工具,诞生于NVIDIA公司,是一

    2023年04月14日
    浏览(42)
  • 深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等

    本文参考了许多优秀博主的博客,大部分安装步骤可在其他博客中找到,鉴于我本人第一次安装后,时隔半年,我忘记了当时安装的许多细节和版本信息,所以再一次报错时,重装花费了大量时间。因此,我觉得有必要把主要过程记录下来,以便下次需要时快速安装。以下过

    2024年02月17日
    浏览(48)
  • PyTorch深度学习环境安装(Anaconda、CUDA、cuDNN)及关联PyCharm

    Tytorch : Python 机器学习库,基于 Torch ,用于自然语言处理等应用程序 Anaconda :是默认的 python 包和环境管理工具,安装了 anaconda ,就默认安装了 conda CUDA : CUDA 是一种由显卡厂商 NVIDIA 推出的通用并行计算架构,该架构使 GPU 能解决复杂的计算问题,可用来计算深度学习 c

    2024年02月13日
    浏览(46)
  • 动手学DL——环境部署随笔【深度学习】【Anaconda】【CUDA】【PyTorch】【jupyter】

    记录虚拟环境安装部署细节,以备重装。 anaconda 2022.10 windows 版本,https://repo.anaconda.com/archive/ 加入环境变量(3/3) D:anaconda3 D:anacondaScripts D:anacondaLibrarybin 配置虚拟环境,命名为 pytorch 查看当前所有的虚拟环境 激活 pytorch 这个虚拟环境 新环境下包很少,安装科学计算库 检查

    2024年02月13日
    浏览(33)
  • (纯小白向)Windows配置GPU深度学习环境:Cuda+Anaconda+pytorch+Vscode

    目录 一、Cuda和Cudnn下载安装 1.1 确定自己的电脑显卡驱动支持的Cuda版本 1.2 Cuda下载与安装 1.3 Cudnn下载与安装 二、Anaconda下载安装 2.1 下载 2.2 安装 2.3 手动配置环境变量 2.4 测试是否安装成功 三、Pytorch下载安装 3.1 创建conda虚拟环境 3.2 Pytorch下载 四、Vscode下载与环境配置 4.1

    2024年02月05日
    浏览(51)
  • 【新手流程】1小时解决Anaconda+Cuda+Cudnn+Pytorch+Pycharm深度学习环境配置

    1、安装Anaconda 2、安装Cuda+Cudnn 3、安装Pytorch 4、安装Pycharm 5、配置Pycharm环境 点击进入Anaconda官网👉: Anaconda.com官网 https://www.anaconda.com/download/ 如果下载速度太慢可以使用这个镜像链接: Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror Index of /anaconda/a

    2024年02月16日
    浏览(59)
  • ubuntu20.04配置ros noetic和cuda,cudnn,anaconda,pytorch深度学习的环境

    这里介绍下本篇文章的目的,为了方便自己日后在其他主机上搭建环境,也为了帮助遇到相同问题的人。本篇文章主要是解决ubuntu20.04搭建机械臂视觉抓取的环境部署问题。第一个环境了ROS环境,第二个环境是深度学习yolov5的环境。 这里推荐鱼香ros的便携式安装方法,这里感

    2024年02月07日
    浏览(60)
  • 【深度学习环境搭建】Windows搭建Anaconda3、已经Pytorch的GPU版本

    无脑下载安装包安装(自行百度) 注意点: 1、用户目录下的.condarc需要配置(自定义环境的地址(别忘了给文件夹加权限);镜像源) 1、先看你的显卡版本 Win +R - 输入命令nvidia-smi,看你的cuda版本 2.下载离线版本安装包(在线也行,只要你有耐心) https://download.pytorch.org

    2024年02月02日
    浏览(44)
  • windows10系统PYthon深度学习环境安装(Anaconda3、PYthon3.10、CUDA11.6、CUDDN10、pytorch、tensorflow,Pycharm)

    一、 總體说明 1、說明:總體採用https://blog.csdn.net/zhizhuxy999/article/details/90442600方法,部分步驟由於版本變化,進行了調整。 2、基本概念 编程语言/编译器:Python。Python的特点是“用最少的代码干最多的事”。Python 2即在2020年停止更新,所以现在学习Python 3是最好的选择。 P

    2023年04月18日
    浏览(51)
  • Anaconda+PyCharm+PyTorch+Gym深度强化学习环境搭建 送新手直接送进炼丹炉

    需要下载的软件和包: Anaconda Pycharm Python PyTorch gym pygame 直接从官网下载: https://www.anaconda.com https://www.anaconda.com/ 点击Download下载即可。 下载好后,打开安装包,自己选一个安装路径,默认路径也行,放其他盘也行,我安装在D盘下的Anaconda文件夹下。 安装向导一路下一步即可

    2024年02月06日
    浏览(41)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包