win10配置cuda和pytorch

标签：显示   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