欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

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

时间:2021-10-23 11:30:01|栏目:Python代码|点击:

虚拟环境的创建

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

pytorch的gpu版本安装

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

在这里插入图片描述

点击系统信息,选择组件

在这里插入图片描述

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

再在pytorch官网选择对应版本

在这里插入图片描述

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

在这里插入图片描述

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

在这里插入图片描述

如上图所示即为成功

上一篇:pandas apply多线程实现代码

栏    目:Python代码

下一篇:在python 中实现运行多条shell命令

本文标题:pytorch安装及环境配置的完整过程

本文地址:http://www.codeinn.net/misctech/180983.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有