win10配置cuda和pytorch

2021-03-20 02:23

阅读:496

标签:显示   pre   电脑   http   app   路径   dmi   admin   执行   

简介

pytorch是非常流行的深度学习框架。下面是Windows平台配置pytorch的过程。

一共需要安装cuda、pycharm、anancoda、pytorch。

主要介绍cuda和pytorch的安装。

安装cuda

1. 根据自己的显卡,选择合适的cuda版本。

百度输入CUDA,进入官网下载。

技术图片

下载结束后,进行安装。

技术图片

安装结束后,自动弹出此窗口。

技术图片

一路下一步

技术图片

在环境变量中,添加cuda的路径

技术图片

在cmd中验证是否安装成功

技术图片

可显示出CUDA版本,说明安装已成功。

2. 下载pytoch

登陆www.pytorch.org进行选择自己电脑的配置。

技术图片

将下方的命令,复制到cmd中执行,注意需用管理员权限打开cmd

技术图片

技术图片

验证是否安装成功

技术图片

技术图片

从cmd中,进入python环境

import torch
torch.__version__
torch.cuda.is_available()

出现以下输出torch版本和‘true‘,则说明安装成功。

win10配置cuda和pytorch

标签:显示   pre   电脑   http   app   路径   dmi   admin   执行   

原文地址:https://www.cnblogs.com/woxiaosade/p/12312137.html


评论


亲,登录后才可以留言!