先查看自己电脑英伟达CUDA支持的版本信息
- 点帮助——系统信息:
- 或者cmd查一下
下载相关CUDA驱动
-
登录后下载
-
以管理员身份运行安装
-
如果当前版本比新版本小,那就打勾继续下一步,否则把勾去掉:
继续安装
-
测试一下安装成功没
有这个版本就成功了
cudnn
- 把另一个压缩包解压,点进去重命名
- 然后复制到刚刚安装的这个v10.2里面
- 配置环境变量
加这两个就行了
pytorch GPU版本的安装
-
pytorch下载:pytorch.org
-
复制命令执行即可,要是速度慢或者中间报错,就加个镜像源给他
-
安装后测试一下
-
说是少这个,那就装一下,一般都没有提示这个的,应该是我重装系统后这个东西没有了
-
重启
如果输出True,那就是安装成功且可以使用gpu加速了 -
如果是用conda安装,加速可以参考下面链接
- https://zhuanlan.zhihu.com/p/144575524
- https://blog.csdn.net/qq_41936559/article/details/102699082 -
结果
如果输出True,那就是安装成功且可以使用gpu加速了