win10下tensorflow、pytorch相关环境配置

最新推荐文章于 2024-07-11 15:35:27 发布

吾即小灾变

最新推荐文章于 2024-07-11 15:35:27 发布

阅读量1k

点赞数

分类专栏： Yolo 深度学习文章标签： tensorflow c++ python

本文链接：https://blog.youkuaiyun.com/longlong068/article/details/105087705

版权

win10下安装tensorflow-gpu2.0.0

显卡GTX 1660Ti

介于Nvida官网未给出GTX 1660Ti的具体算力，可以参考cuda的维基百科https://en.wikipedia.org/wiki/CUDA

下载CUDA版本： cuda_10.0.130_411.31_win10

链接：https://developer.nvidia.com/cuda-10.0-download-archive
下载cuda10.0对应得cudnn版本： cuDNN v7.6.5

链接：https://developer.nvidia.com/rdp/cudnn-download，直接下载较慢，建议右键复制链接采用迅雷下载。
安装CUDA，此部分操作详细见https://blog.youkuaiyun.com/abc13526222160/article/details/88780278。

在cmd中输入 nvcc -V查看cuda是否安装成功。

配置cudnn，此部分操作按https://blog.youkuaiyun.com/weixin_43318717/article/details/94433790。

Cudnn文件夹下的: bin/cudnn64_7.dll	拷贝	C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
Cudnn文件夹下的: include/cudnn.h	拷贝	C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
Cudnn文件夹下的: lib/x64/cudnn.lib	拷贝	C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64

安装tensorflow-gpu

activate python36 #激活python36这个环境
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple #GPU版本
pip install tensorflow==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple #CPU版本
#验证
import tensorflow as tf
print(tf.test.is_gpu_available