安装cuda10.2及对应cuDNN,安装pytoch1.10至anaconda3
安装cuda10.2及对应cudNN,安装pytoch至anaconda
刚开始做毕设,down下来的模型需要使用pytorch框架,然后在pycharm 的IDE中下载的版本只能是简单的CPU版本,无法调用GPU,所以在跑模型的时候发现无法调用cuda,所以只能重装pytorch,而这一重装就发现了很多问题。
1.查看NVIDA支持的CUDA版本
打开NVIDA控制面板
点击上方 帮助(H)->系统信息->组件
在这里可以看到你所支持的最高cuda版本,我的是11.5.107,然后
原创
2021-11-23 23:23:57 ·
1723 阅读 ·
2 评论