cuda安装精简

使用torch的一定注意cuda版本必须要跟torch版本对应,网上好多网友写的博客都没提到,目前cuda官网的版本是11.0但是我当时在torch官网下载的是支持cuda10.2版本的torch。 cuda 10.2 链接是cuda10.2版本的下载链接,下载之后按照提示步骤一步步安装即可非常简单。 两种方法测试是否安装成功: 1.进入cmd,输入nvcc -V,会出现cuda的版本号等相关信息
相关文章
相关标签/搜索