深度学习环境配置
风小子 哦
22年计科新生,不定时更新,分享学习过程中遇到的一些问题以及对应的解法,希望能和各位共同进步。
展开
-
torch_scatter ,torch-sparse,torch-cluster等系列安装
torch_scatter ,torch-sparse,torch-cluster等系列安装原创 2024-04-22 21:29:29 · 931 阅读 · 0 评论 -
正确查看自己的CUDA版本(不要再被骗啦)
CUDA分为两种,驱动API和运行API,驱动API指的是指的显卡驱动支持的最高cuda版本,我们运行程序时用的是运行API。nvidia-smi显示的是驱动所能支持的最大运行API版本,nvcc --version查看的是CUDA的运行API版本。不要再傻傻地以为这就是你的cuda版本,然后再下载对应的dgl和pytorch了,不然当你配环境时你会配到怀疑人生(笔者亲自踩的坑)。原创 2023-10-22 13:17:28 · 4827 阅读 · 8 评论 -
Tensorflow的gpu无法使用,明明安装了却用不了
2023年现在在安装tensorflow时,按照官网的直接pip install tensorflow,就算你安装后TensorFlow的GPU 也无法使用。你明明就看见他官网的注释下载的就是CPU 和 GPU 的当前稳定版本,但就是报错用不了。原创 2023-12-27 16:02:06 · 1081 阅读 · 1 评论