MX150 WIN10 安装TensorFlow-GPU

需要 4 个文件

已经安装了python3.6.5
在这里插入图片描述

  1. 419.67-notebook-win10-64bit-international-whql.exe
    驱动程序 | GeForce
  2. tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl
    fo40225/tensorflow-windows-wheel
  3. cuda_9.2.148_win10_network.exe
    CUDA Toolkit Archive
  4. cudnn-9.2-windows10-x64-v7.1.zip
    cuDNN Download (需要注册)

可用cmd的nvidia-smi命令检查GPU状态,要保证此命令不出错。

若出现:nvidia-smi显示不是内部或外部命令也不是可运行的程序
则把C:\Program Files\NVIDIA Corporation\NVSMI添加到path,再重新打开cmd窗口!

DriverPack-17-Online.exe
这个软件也有用,把显卡卸载掉后用这个恢复驱动什么的。
在这里插入图片描述

在这里插入图片描述

参考

在nvidia-mx150的win10安裝cuda-toolkit-cudnn-python-anaconda-and-tensorflow

深度学习_tensorflow-gpu for Win10-64bit安装踩坑流程(cuda9.2 ,cudnn7.1)

补充

tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl还有1.7的某个版本(尝试过的)都存在问题:tf.pow()会出现nan空值!
tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl不会出现这个问题,但是会出现

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2

大概意思是:你的CPU支持AVX扩展,但是你安装的TensorFlow版本无法编译使用。在此强调一下:
这只限于CPU。如果你有一个GPU,你不应该关心AVX的支持,因为大多数昂贵的操作将被分派到一个GPU设备上(除非明确地设置)。

原因:除了通常的算术和逻辑,现代CPU提供了许多低级指令,称为扩展,例如, SSE2,SSE4,AVX等。高级矢量扩展(AVX)是英特尔在2008年3月提出的英特尔和AMD微处理器的x86指令集体系结构的扩展,英特尔首先通过Sandy Bridge处理器在2011年第一季度推出,随后由AMD推出Bulldozer处理器在2011年第三季度.AVX提供了新功能,新指令和新编码方案。特别是,AVX引入了融合乘法累加(FMA)操作,加速了线性代数计算,即点积,矩阵乘法,卷积等。几乎所有机器学习训练都涉及大量这些操作,因此将会支持AVX和FMA的CPU(最高达300%)更快。该警告指出您的CPU确实支持AVX(hooray!)。

在这种情况下,您可以简单地忽略此警告。

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

TF_CPP_MIN_LOG_LEVEL 取值 00也是默认值,输出所有信息
TF_CPP_MIN_LOG_LEVEL 取值 1 : 屏蔽通知信息
TF_CPP_MIN_LOG_LEVEL 取值 2 : 屏蔽通知信息和警告信息
TF_CPP_MIN_LOG_LEVEL 取值 3 : 屏蔽通知信息、警告信息和报错信息

tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64还会出现如下问题:
在这里插入图片描述
tensorflow_gpu-1.10.0-cp36-cp36m-win_amd64.whl
Great!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
安装PyTorch的GPU版本,首先需要确保你的电脑具备支持CUDA的显卡。根据引用中提到的,你的电脑使用的是GeForce MX150显卡,它是支持CUDA的。 接下来,需要按照以下步骤安装PyTorch的GPU版本: 1. 首先,下载并安装适用于你的CUDA版本的CUDA Toolkit。CUDA Toolkit是一个用于支持GPU计算的软件开发工具包。你可以从NVIDIA官方网站上下载与你的CUDA版本相对应的Toolkit。 2. 安装CUDA Toolkit后,打开Anaconda Prompt,输入以下命令来创建一个新的虚拟环境(可选步骤): ``` conda create --name myenv ``` 其中,"myenv"是你给虚拟环境起的名字,你可以根据自己的需求进行命名。 3. 激活刚才创建的虚拟环境: ``` conda activate myenv ``` 4. 接下来,使用conda命令来安装PyTorch的GPU版本。根据引用中提到的,选择适合你的PyTorch和torchvision版本。例如,如果你想要安装PyTorch的1.4.0版本和torchvision的0.5.0版本,可以使用以下命令: ``` conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=<your_cuda_version> -c pytorch ``` 其中,"<your_cuda_version>"是你所安装的CUDA Toolkit的版本号。 5. 等待安装完成后,就成功安装了PyTorch的GPU版本。 总结一下安装PyTorch的GPU版本的步骤: 1. 确保你的电脑具备支持CUDA的显卡。 2. 下载并安装适用于你的CUDA版本的CUDA Toolkit。 3. (可选步骤)创建一个新的虚拟环境。 4. 激活虚拟环境。 5. 使用conda命令安装PyTorch的GPU版本。 希望这些步骤对你有帮助!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值