GPU
北京纯牛奶
做个知识点记录的工具而已,其他绕道!
展开
-
查看显卡和cuda是否安装成功
我被这个问题尴尬了很久,nvidia-smi 查看显卡驱动是否安装成功nvcc -V 查看cuda是否安装成功,其实也就是cuda是否安装成功记住这两个都需要网上下载对应版本的软件,下载之后正常跟安装其他的软件一样安装就可以,但是cuda安装的时候,一会也会涉及安装pytorch,别去最新官网安装了,因为你已经安装了cuda了,所以找对应版本的安装就可以,网址:https://pytorch.org/get-started/previous-versions/原创 2021-12-05 21:08:32 · 4015 阅读 · 0 评论 -
本机cuda和环境cuda版本不对应导致没办法调用
有的的时候,自己新建的环境和本地安装的cuda版本不对应,会导致自己的没办法使用GPU运行nvcc -V 显示的是本机的cuda版本,torch.version.cuda 显示的是该环境的版本,记得在import torch 里面进行输入因为两个版本需要对应,最好就是本机安装什么cuda版本,环境就安装什么cuda版本,目前我是这样解决的...原创 2021-12-05 21:00:30 · 1929 阅读 · 4 评论