使用torch的一定注意cuda版本必须要跟torch版本对应,网上好多网友写的博客都没提到,目前cuda官网的版本是11.0但是我当时在torch官网下载的是支持cuda10.2版本的torch。
cuda 10.2 链接是cuda10.2版本的下载链接,下载之后按照提示步骤一步步安装即可非常简单。
两种方法测试是否安装成功:
1.进入cmd,输入nvcc -V
,会出现cuda的版本号等相关信息。
2.如果正在用torch的话,直接输torch.cuda.is_available()
,如果返回True
则表示cuda安装成功,我们可以用GPU训练神经网络。