Windows+pycharm安装pytorch

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