• Win11配置多个CUDA环境


    概述

    由于跑项目发现需要配置不同版本的Pytorch,而不同版本的Pytorch又对应不同版本的CUDA,于是有了在Win上装多个CUDA的打算

    默认已经在电脑上装了一个CUDA

    屏幕截图 2023-09-30 112423

    现在开始下载第二个CUDA版本,前面下载的操作和普通安装的几乎一样

    安装CUDA

    CUDA下载链接

    下载自己需要的CUDA版本,以我自己要安装的为例

    屏幕截图 2023-09-30 095614

    屏幕截图 2023-09-30 095815

    下载打开exe文件进行安装,第一步不用管,直接点击确认

    屏幕截图 2023-09-30 100402

    后面选择自定义安装,只选择安装CUDA

    屏幕截图 2023-09-30 100534

    其他一路下一步就可以了

    安装cuDNN

    cuDNN下载链接

    PS:下载需要登录一下账号

    屏幕截图 2023-09-30 100047

    下载下来后,把压缩包解压,得到如下文件

    image-20230930100746137

    bin,include文件夹中的文件,分别复制到下列地址对应的文件夹下(CUDA的安装地址,以我的地址为例),遇到提示,为全部选择覆盖和替换。

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7
    
    • 1

    需要注意的是,lib文件夹里的所有文件,需要复制到

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\lib\x64
    
    • 1

    至此,安装工作就做完了,可以看到电脑上已经有两个CUDA版本了

    image-20230930101157014

    配置环境

    由于CUDA11.7是后面安装的,所以可以看到系统把之前的环境覆盖了

    屏幕截图 2023-09-30 103109

    以作者在为例,系统变量的PATH里配置一下路径

    CUDA_11.7

    %CUDA_PATH_V11_7%\lib\x64
    %CUDA_PATH_V11_7%\include
    %CUDA_PATH_V11_7%\extras\CUPTI\lib64
    %CUDA_PATH_V11_7%\bin
    %CUDA_PATH_V11_7%\libnvvp
    
    • 1
    • 2
    • 3
    • 4
    • 5

    CUDA_11.0

    %CUDA_PATH_V11_0%\lib\x64
    %CUDA_PATH_V11_0%\include
    %CUDA_PATH_V11_0%\extras\CUPTI\lib64
    %CUDA_PATH_V11_0%\bin
    %CUDA_PATH_V11_0%\libnvvp
    
    • 1
    • 2
    • 3
    • 4
    • 5

    配置好如下:

    微信截图_20230930105036

    配置好之后,默认CUDA版本是11.7的,如果要切换到11.0版本的,只需要把下面的五条整体上移到CUDA_11.7配置的前面

    微信截图_20230930105411

    验证

    测试CUDA11.7

    首先验证CUDA_11.7是否配置成功

    image-20230930105601201

    测试Pytorch是否可以调用显卡

    python
    import torch
    torch.cuda.is_available()
    torch.cuda.get_device_name(0)
    torch.version.cuda
    
    • 1
    • 2
    • 3
    • 4
    • 5

    image-20230930120418607

    测试CUDA11.0

    先把系统变量里Path里相关路径上移

    微信截图_20230930110409

    然后重启!重启!重启!

    验证CUDA_11.7是否配置成功

    image-20230930112425329

    测试Pytorch是否可以调用显卡

    image-20230930112626663

    总结

    切换CUDA的时候除了要修改系统变量Path中的路径顺序,一定要记得重启!

  • 相关阅读:
    验证了一遍CVAT的安装(Windows 11)
    DDD-如何集成限界上下文和应用服务的使用
    Docker搭建一个Wordpress博客
    C语言序列化与反序列化--TCL中支持的数据结构(二)
    非零基础自学Java (老师:韩顺平) 第13章 常用类 13.11 日期类
    error: undefined reference to ‘vtable for …’
    Python 实现http server接收mutipart/form-data文件 方法1
    机器学习--决策树(sklearn)
    Rails 中的布局和渲染
    LeetCode(17)罗马数字转整数【数组/字符串】【简单】
  • 原文地址:https://blog.csdn.net/ruvikm/article/details/133430914