pytorch安装及环境配置的完整过程
虚拟环境的创建
命令行窗口中使用
conda create -n 环境名 python=所需python版本
即可创建虚拟环境
pytorch的gpu版本安装
首先确定自己电脑的gpu版本
打开显卡控制面板

点击系统信息,选择组件

产品名称中CUDA后的11.0便是gpu版本
再在pytorch官网选择对应版本

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

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

如上图所示即为成功
到此这篇关于pytorch安装及环境配置的完整过程的文章就介绍到这了,更多相关pytorch安装及环境配置内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
相关文章
Python数据分析之双色球统计单个红和蓝球哪个比例高的方法
这篇文章主要介绍了Python数据分析之双色球统计单个红和蓝球哪个比例高的方法,涉及Python数值运算及图形绘制相关操作技巧,需要的朋友可以参考下2018-02-02
详解使用python的logging模块在stdout输出的两种方法
这篇文章主要介绍了详解使用python的logging模块在stdout输出的相关资料,需要的朋友可以参考下2017-05-05


最新评论