概述
准备工作
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正式安装所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复