Advertisement

Windows10下CUDA及cuDNN、pytorch及torchvision在第二部分中进行安装

阅读量:

前言:

前一篇博客介绍了CUDA与cuDNN的安装步骤,鉴于CUDA、cuDNN、pytorch及torchvision之间的版本需要相互兼容,本文将重点阐述pytorch与torchvision的安装流程。

当前我的系统配置如下: Windows10操作系统 + Anaconda3环境 + Python 3.8版本 + CUDA10.2 + cuDNN

关于安装前的版本匹配建议: CUAD 10.2对应的pytorch版本为1.6.0,而torchvision则应选择0.7.0版本。

在这里插入图片描述

该信息来源于pytorch官方下载页面,若需查找其他版本对应的适配方案,可参考此网站:https://pytorch.org/get-started/previous-versions/

一、pytorch, torchvision的安装

1. 创建pytorch环境

1)启动命令提示符(cmd),执行命令:conda --version,以查看当前安装的Anaconda版本信息。随后输入:`cond

全部评论 (0)

还没有任何评论哟~