我是靠谱客的博主 还单身母鸡,最近开发中收集的这篇文章主要介绍windows安装pytorch以及tensflow正式安装,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

准备工作

1.安装anaconda比较简单,一直下一步就好。

2.下载cuda,cudnn

        注意:cuda,cudnn版本和自身电脑的显卡要匹配

        在官网这里可以搜到显卡对应cuda的适配版本,然后去下载cuda

然后在这里找到与cuda版本对应的cudnn下载 ,如下图

不过cudnn下载需要进行注册,不想注册的话可以直接点鼠标右键复制链接然后用迅雷之类的下载工具下载

        附上查看英伟达显卡计算能力表https://developer.nvidia.com/zh-cn/cuda-gpus#compute

正式安装

1.安装cuda和cudnn

        找到刚才下载好的cuda所在文件夹,双击应用程序进行安装,可以选择精简安装(不可以设置安装位置)也可以自定义安装,基本上都是一直下一步就好

        安装完之后按windows键+e键 然后输入cmd ,进入如下界面

        输入nvcc -V,判断是否cuda安装成功

 

 cuda安装成功之后,解压下载的cudnn,将cudnn文件夹里面的这三个文件复制到cuda的安装目录里面,就可以了

 

2.安装pytorch

方法一:

按windows键 打开anaconda prompt

输入 conda create -n pytorch python=3.6(这个3.6是安装的python版本),

conda create -n pytorch python=3.6

然后输入y确认安装就好

装好之后输入conda activate pytorch进入pytorch

conda activate pytorch

 

然后进入pytorch官网 向下滑到这里复制这串代码  按回车 就可以自动安装了(如果这个推荐的版本不合适可以进入下面那个箭头指的地方找别的版本)

这个方法可能因为网络原因下载不了,或者下载的比较慢,可以使用pycharm 安装

方法二:

去官网下载pytorch,这次选择pip,然后复制红色框里的链接

 再这个页面中找到你要安装的pytorch版本下载

 接下来

打开pycharm新建一个项目

然后新建一个python文件

 输入以下代码运行,

import torch

print(f"Hello PyTorch {torch.__version__}")

  然后回报错,接下来点击Terminal,输入cd +刚才下载的pytorch所在目录,

 

 输入pip install 然后按住tab键会自动补上pytorch的文件名,如果第一个文件不是的话继续按tab键切换,然后按回车等它自己安装就好了

 安装好之后再次运行就正常了

3.安装tensorflow

 首先,配置tensorflow环境

打开anaconda prompt 输入下面代码

conda create -n tensorflow_gpu python=3.6

等一会再输入y确定就好

然后再切换到 tensorflow_gpu环境中,输入下面代码

activate tensorflow_gpu

输入代码,安装anaconda包

conda install anaconda

 接下来开始正式安装tensorflow

注意conda install tensorflow-gpu这个中间的横线是"-"不是"_"

conda install tensorflow-gpu

然后tensorflow就安装好了

最后

以上就是还单身母鸡为你收集整理的windows安装pytorch以及tensflow正式安装的全部内容,希望文章能够帮你解决windows安装pytorch以及tensflow正式安装所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(44)

评论列表共有 0 条评论

立即
投稿
返回
顶部