您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:pytorch安装及环境配置的完整过程

51自学网 2021-10-30 22:49:04
  python
这篇教程pytorch安装及环境配置的完整过程写得很实用,希望能帮到您。

虚拟环境的创建

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

pytorch的gpu版本安装

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

在这里插入图片描述

点击系统信息,选择组件

在这里插入图片描述

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

再在pytorch官网选择对应版本

在这里插入图片描述

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

在这里插入图片描述

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

在这里插入图片描述

如上图所示即为成功

到此这篇关于pytorch安装及环境配置的完整过程的文章就介绍到这了,更多相关pytorch安装及环境配置内容请搜索51zixue.net以前的文章或继续浏览下面的相关文章希望大家以后多多支持51zixue.net!


Python 实现图像逐像素点取邻域数据
基于Python的XML格式的文件示例代码详解
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。