首页 > 编程开发 > Python    日期:2021-09-30 / 浏览

虚拟环境的创建

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

pytorch的gpu版本安装

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

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

点击系统信息,选择组件

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

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

再在pytorch官网选择对应版本

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

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

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

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

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

如上图所示即为成功

觉得上面的内容有用吗?快来点个赞吧!

点赞() 我要打赏

温馨提示 : 本站内容来自会员投稿以及互联网,所有源码及教程均为作者总结编辑,请大家在使用过程中提前做好备份,以免发生无法预知的错误,源码类教程请勿直接用于生产环境!

 可能感兴趣的文章

1 2 3 4 5