WIN10 + GTX1660Ti配置TensorFlow GPU版本

本文详细介绍如何在Windows10系统上,使用Anaconda安装Python3.7.3,并配置TensorFlow-GPU1.14.0、Cuda10.0及Cudnn7.6.1,包括下载源、安装步骤及测试方法。

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

安装准备:

         操作系统: Windows 10
  
       Python版本: 3.7.3
        安装Anaconda3-2019.03-Windows-x86_64自带python3.7.3
        下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
        参考链接:https://blog.youkuaiyun.com/junchengberry/article/details/81217645
        
        TensorFlow-GPU版本: 1.14.0
        
        Cuda版本: cuda_10.0.130_411.31_win10
        下载地址:https://developer.nvidia.com/cuda-10.0-download-archive
        
        Cudnn版本: cudnn-10.0-windows10-x64-v7.6.1.34.zip
        下载地址:https://developer.nvidia.com/rdp/cudnn-download
       
       清华镜像:
       -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

       conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
       conda config --set show_channel_urls yes

安装过程:

1.安装python
安装anaconda即可

2.安装cuda
在系统变量环境变量中path配置(已经有的不用再次配置):
在这里插入图片描述
在cmd命令窗口中输入

nvcc -V

查看cuda是否安装成功:
在这里插入图片描述
3. 安装Cudnn,解压cudnn-9.0-windows10-x64-v7.zip
将Cudnn包中所带的文件放到Cuda目录(上面的CUDA_PATH的路径)下对应的文件夹即可
Cudnn文件夹下的: bin/cudnn64_7.dll 拷贝到 Cuda文件夹下的 bin/ 目录下
Cudnn文件夹下的: include/cudnn.h 拷贝到Cuda文件夹下的 include/ 目录下
Cudnn文件夹下的: lib/x64/cudnn.lib 拷贝到Cuda文件夹下的 lib/x64/cudnn.lib目录下
(cuda默认安装路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0)

4.安装tensorflow-gpu

pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

5.测试

  import tensorflow as tf
    a = tf.constant(1)
    b = tf.constant(2)
    sess = tf.Session()
    print(sess.run(a + b))

输出为 3

<think>嗯,用户想了解如何开启VMware的GPU直通,也就是PCI Passthrough。首先,我需要回忆一下VMware的相关知识。记得PCI Passthrough允许虚拟机直接访问物理硬件设备,比如GPU,这样可以提升图形性能,适合需要GPU加速的应用。不过,这个过程可能涉及多个步骤,需要确保硬件和软件的支持。 首先,应该检查硬件是否支持VT-d或AMD-Vi,这是Intel和AMD的虚拟化技术,用于IOMMU。然后,主板的BIOS/UEFI里需要开启这些选项。用户可能不太清楚怎么进入BIOS设置,可能需要指导他们重启电脑并按特定键(比如Del或F2)进入设置,找到相关选项启用。 接下来是VMware ESXi的情况,因为用户可能使用的是ESXi而不是Workstation。ESXi中配置PCI Passthrough需要在主机管理界面里将GPU标记为可直通设备,然后重启主机。之后在虚拟机的设置中添加该PCI设备。不过,不同的VMware产品步骤可能不同,比如Workstation和ESXi可能有差异,需要确认用户的具体环境。 另外,需要注意GPU是否被宿主机占用,如果宿主机已经在使用GPU,可能无法直通。这时候可能需要配置ESXi使用集成显卡,或者使用没有显示输出的GPU。还有驱动问题,虚拟机安装对应的GPU驱动也是必须的,否则即使直通成功,也可能无法正常工作。 还有一点,某些GPU可能不支持直通,尤其是消费级的显卡,厂商可能限制了其在虚拟化环境中的使用。比如NVIDIA的GeForce系列和Quadro/Tesla系列的区别。用户可能需要确认他们的GPU型号是否支持。 可能用户会遇到的问题包括直通后虚拟机无法启动,或者设备冲突。这时候需要检查日志,比如ESXi的vmkernel.log,查看具体的错误信息。另外,是否需要配置虚拟机为EFI引导,或者调整其他高级设置。 总结一下,步骤大概是:检查硬件支持、启用BIOS设置、配置ESXi的PCI Passthrough、重启主机、添加设备到虚拟机、安装驱动。不过要分点详细说明,确保用户能够逐步操作,并提醒可能遇到的坑。</think>开启 VMware GPU 直通(PCI Passthrough)需分步骤完成,以下是详细操作指南: --- ### **一、前提条件检查** 1. **硬件支持** - CPU 需支持 **VT-d(Intel)** 或 **AMD-Vi(AMD)** 技术 - 主板 BIOS/UEFI 需开启 `Intel VT-d`/`AMD SVM` 和 `IOMMU` - GPU 需支持直通(建议查阅 VMware 兼容性列表) 2. **软件环境** - 使用 **VMware ESXi**(企业级虚拟化平台) - 宿主机系统版本需支持 PCI Passthrough(ESXi 6.5+ 已验证) --- ### **二、BIOS/UEFI 设置** 1. 重启服务器/主机,进入 BIOS 设置界面 2. 启用以下选项(不同主板名称可能略有差异): ```plaintext - Intel VT-d / AMD IOMMU - SR-IOV(可选,用于多虚拟机共享 GPU) - 禁用 GPU 的 UEFI 安全启动(部分 NVIDIA GPU 需要) ``` --- ### **三、ESXi 主机配置** 1. **标记 GPU 为可直通设备** - 登录 ESXi Web Client → 导航至 `主机 → 管理 → 硬件 → PCI 设备` - 找到目标 GPU(如 `NVIDIA Corporation GP102 [GeForce GTX 1080 Ti]`) - 点击 **切换直通**,状态变为 `活动` 表示已启用 2. **重启 ESXi 主机** - 直通配置需重启生效 --- ### **四、虚拟机配置** 1. **编辑虚拟机设置** - 关闭虚拟机 → 进入 `虚拟机选项 → 高级 → 编辑配置` - 添加参数:`hypervisor.cpuid.v0 = FALSE`(绕过 NVIDIA 驱动限制) 2. **添加 PCI 设备** - 在虚拟机硬件设置中点击 `添加其他设备 → PCI 设备` - 选择已直通的 GPU 3. **调整虚拟机操作系统类型** - 建议选择 `Linux` 或 `Windows` 的最新版本(确保驱动兼容性) --- ### **五、安装 GPU 驱动** 1. **在虚拟机中安装操作系统** 2. **下载对应 GPU 厂商的官方驱动** - NVIDIA:需使用 **vGPU 驱动** 或 修改版驱动(消费级 GPU) - AMD:建议使用 `AMDVI` 兼容驱动 3. **禁用宿主机对 GPU 的占用** - 若宿主机使用该 GPU 输出显示,直通可能失败(需改用集成显卡) --- ### **六、验证排错** 1. **验证直通状态** - 在虚拟机中运行 `lspci`(Linux)或 `设备管理器`(Windows)检查 GPU 是否识别 2. **常见问题** - **错误:`设备被占用`** 确保 ESXi 主机未使用该 GPU 输出显示 - **错误:`Failed to power on VM`** 检查 `hypervisor.cpuid.v0` 参数是否正确,或尝试禁用 EFI 安全启动 --- ### **注意事项** - **性能损耗**:直通后 GPU 由虚拟机独占,宿主机无法使用 - **GPU 兼容性**:NVIDIA 消费级显卡可能限制虚拟化功能(需修改驱动或使用专业卡) - **热插拔**:PCI Passthrough 不支持热插拔,修改配置需关闭虚拟机 --- 通过以上步骤,可完成 VMware GPU 直通的配置。若需多虚拟机共享 GPU,需改用 **SR-IOV** 或 **vGPU** 技术(需特定硬件支持)。
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值