最近要参加一个比赛,需要用到深度学习的框架,老师建议我们用pytorch,但是安装的时候总是发现各种错误。
解决方案是:去pytorch官网生成下载的命令,然后再cmd上执行

这里我们发现有CUDA的选项,这就需要我们安装cuda
怎么安装cuda?
首先我们要知道我们的独立显卡的型号:

然后我们要知道我们的显卡是否支持cuda:点击这里
发现是支持的,算力为6.1:

接下来就要下载cuda了,选择10.1(最好)或者10.0都行:
在参加深度学习比赛时,根据老师建议使用PyTorch。本文介绍了如何在Windows上通过PyCharm安装PyTorch,包括检查CUDA支持、下载安装CUDA 10.1,配置环境变量,下载CUDNN并创建账户,最后成功安装PyTorch并验证安装效果。
最近要参加一个比赛,需要用到深度学习的框架,老师建议我们用pytorch,但是安装的时候总是发现各种错误。
解决方案是:去pytorch官网生成下载的命令,然后再cmd上执行

这里我们发现有CUDA的选项,这就需要我们安装cuda
怎么安装cuda?
首先我们要知道我们的独立显卡的型号:

然后我们要知道我们的显卡是否支持cuda:点击这里
发现是支持的,算力为6.1:

接下来就要下载cuda了,选择10.1(最好)或者10.0都行: