虚拟环境的创建
命令行窗口中使用
conda create -n 环境名 python=所需python版本
即可创建虚拟环境
pytorch的gpu版本安装
首先确定自己电脑的gpu版本
打开显卡控制面板
点击系统信息，选择组件
产品名称中CUDA后的11.0便是gpu版本
再在pytorch官网选择对应版本
进入为pytorch创建的虚拟环境，输入命令，等待片刻，pytorch便安装完成了。
再在NVIDIA官网安装cuda和cudnn
将cudnn的三个文件分别放入cuda安装目录下，即可完成配置
在命令行中输入验证代码
如上图所示即为成功
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
评论(0)