注意 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就安装好了
