win10 NVIDIA GeForce GTX 950M安装tensorflow_gpu_1.9.0、torch

迫于计算需求，最近将笔记本上的GPU也利用了起来。CUDA+CuDNN+TensorFlow部分参考了博文，感谢博主让我少走了很多弯路。Torch部分尝试了多种方法未果，最后凭借经验完成了torch1.1.0和torchvision0.4.1的安装。

硬件配置

显卡：NVIDIA GeForce GTX 950M
Python：3.6.2
CUDA：9.0 -> cuda_9.0.176_win10
CuDNN：9.0 -> cudnn-9.0-windows10-x64-v7.6.5.32
参考链接：https://blog.csdn.net/weixin_43741442/article/details/84260575

安装TensorFlow

pip install tensorflow_gpu==1.9.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

检验TensorFlow是否安装成功

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

运行结果

安装PyTorch

pip install http://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl

安装torchvision

下载torchvision-0.4.1-cp36-cp36m-win_amd64.whl，下载链接
安装torchvision，使用命令pip install torchvision-0.4.1-cp36-cp36m-win_amd64.whl

检验是否安装成功

import torch
torch.cuda.is_available()   # 检验GPU是否可用
import torchvision

安装paddlepaddle

python -m pip install paddlepaddle-gpu==1.5.1.post97 -i https://mirror.baidu.com/pypi/simple

验证是否安装成功

import paddle.fluid as fluid
fluid.install_check.run_check()

至此，三大深度学习框架安装成功！

win10 NVIDIA GeForce GTX 950M安装tensorflow_gpu_1.9.0、torch_1.1.0和paddlepaddle1.5.1相关推荐

win10+NVIDIA geforce gtx 960M+cuda9.0+cudnn7.1+pytorch_1.1.0+torchvision_0.3.0---已解决
今天折腾了一天,终于是把它折腾出来了,可能真的是自己太菜了,废话不多说,简单记录一下. 配置系统 :win10 python版本 :3.7.3 显卡 :NVIDIA geforce gtx 960M ...
NVIDIA GeForce GTX 950M 新出驱动程序
1.先上官网链接NVIDIA 引领人工智能计算 | NVIDIA 2.点击 "驱动程序" 接下来就开始选取你自己电脑对应的显卡版本.选取注意项: ①产品类型:查询方式:win+R ...
windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0
安装 Visual Studio 2019下载地址下载完成后点击exe,勾选通用Windows平台开发和使用c++的桌面开发,直到安装完成. 右键点击NVIDIA控制面板,进入帮助->系统信息 ...
NVIDIA GeForce GTX 850M win10 cuda配置及安装教程
NVIDIA GeForce GTX 850M win10 cuda配置 cuda10.0+cudnn7.6.4+tensorflow2.0.0+keras2.3.0 ---------------- ...
NVIDIA GeForce GTX 1060 驱动安装--------TensorFlow系列学习笔记（一）
NVIDIA GeForce GTX 1060 驱动安装 1. 英伟达官网下载驱动程序 2. 禁用 nouveau 第三方驱动 3. 安装 NVIDIA GeForce GTX 1060 驱动 NVI ...
MAC OS X10.9.5下成功驱动独立显卡影驰Nvidia GeForce GTX 760大将(4GB)--非公版
我的机器是台式机(自己组装的),硬件参数如下: 操作系统 Windows 7 旗舰版 64位 SP1 ( DirectX 11 )处理器英特尔 Core i7-4770K @ 3.50GHz 四核 ...
Nvidia GeForce GTX 1650不支持OpenGL4.6
目录 1 更新Nvidia GeForce GTX 1650显卡驱动到最新的NVIDIA 512.15 2 安装cuda_11.5.1_496.13_windows.exe后,仍然不支持openg ...
Intel graphics HD 4600 核芯显卡对比 nVIDIA Geforce GTX 750 2GB GDDR5 独立显卡
Intel graphics HD 4600 核芯显卡对比 nVIDIA Geforce GTX 750 2GB GDDR5 独立显卡参数比较,3D Mark 性能测试下面列出的这些比较项目与参 ...
linux64x gtx970,Nvidia GeForce GTX 970 ( 4 GB / 七彩虹 )无法正常驱动
先上问题图片说明每次启动mac都要按空格然后选定boot macos with selected options然后回车才能正常驱动显卡现在上电脑配置电脑型号技嘉 G1.Sniper B ...

win10 NVIDIA GeForce GTX 950M安装tensorflow_gpu_1.9.0、torch_1.1.0和paddlepaddle1.5.1

硬件配置

安装TensorFlow

检验TensorFlow是否安装成功

运行结果

安装PyTorch

安装torchvision

检验是否安装成功

安装paddlepaddle

验证是否安装成功

win10 NVIDIA GeForce GTX 950M安装tensorflow_gpu_1.9.0、torch_1.1.0和paddlepaddle1.5.1相关推荐

最新文章

热门文章