win10配置cuda和pytorch

Posted 2021-03-09 woxiaosade

简介

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‘，则说明安装成功。