最近要参加一个比赛,需要用到深度学习的框架,老师建议我们用pytorch,但是安装的时候总是发现各种错误。
解决方案是:去pytorch官网生成下载的命令,然后再cmd上执行
这里我们发现有CUDA的选项,这就需要我们安装cuda
怎么安装cuda?
首先我们要知道我们的独立显卡的型号:
然后我们要知道我们的显卡是否支持cuda:点击这里
发现是支持的,算力为6.1:
接下来就要下载cuda了,选择10.1(最好)或者10.0都行:
最近要参加一个比赛,需要用到深度学习的框架,老师建议我们用pytorch,但是安装的时候总是发现各种错误。
解决方案是:去pytorch官网生成下载的命令,然后再cmd上执行
这里我们发现有CUDA的选项,这就需要我们安装cuda
怎么安装cuda?
首先我们要知道我们的独立显卡的型号:
然后我们要知道我们的显卡是否支持cuda:点击这里
发现是支持的,算力为6.1:
接下来就要下载cuda了,选择10.1(最好)或者10.0都行: