这篇文章主要介绍了pytorch安装及环境配置的完整过程,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
虚拟环境的创建
命令行窗口中使用conda create -n 环境名 python=所需python版本
即可创建虚拟环境
pytorch的gpu版本安装
首先确定自己电脑的gpu版本
打开显卡控制面板
点击系统信息,选择组件
产品名称中CUDA后的11.0便是gpu版本
再在pytorch官网选择对应版本
进入为pytorch创建的虚拟环境,输入命令,等待片刻,pytorch便安装完成了。
再在NVIDIA官网安装cuda和cudnn
将cudnn的三个文件分别放入cuda安装目录下,即可完成配置
在命令行中输入验证代码
如上图所示即为成功
以上就是pytorch安装及环来源gaodaimacom搞#^代%!码&网境配置的完整过程的详细内容,更多请关注gaodaima搞代码网其它相关文章!