pytorch安装及环境配置 pytorch安装及环境配置的完整过程

水煎茶 2021-03-17 我要评论

想了解pytorch安装及环境配置的完整过程的相关内容吗，水煎茶在本文为您仔细讲解pytorch安装及环境配置的相关知识和一些Code实例，欢迎阅读和指正，我们先划重点：pytorch安装及环境配置,pytorch环境配置，下面大家一起来学习吧。

虚拟环境的创建

命令行窗口中使用
conda create -n 环境名 python=所需python版本
即可创建虚拟环境

首先确定自己电脑的gpu版本
打开显卡控制面板

在这里插入图片描述

点击系统信息，选择组件

在这里插入图片描述

产品名称中CUDA后的11.0便是gpu版本

再在pytorch官网选择对应版本

在这里插入图片描述

进入为pytorch创建的虚拟环境，输入命令，等待片刻，pytorch便安装完成了。
再在NVIDIA官网安装cuda和cudnn

在这里插入图片描述

将cudnn的三个文件分别放入cuda安装目录下，即可完成配置
在命令行中输入验证代码

在这里插入图片描述

如上图所示即为成功