Linux服务器有四块GPU但torch.cuda.device_count(）找不到

pip list 结果

conda list 结果

我的cuda版本是11.1

在此之前，建议大家按照这个教程 把  /.bashrc 中的环境配好（即添加cuda到/.bashrc环境变量）

vim ~/.bashrc

 打开之后 :w 进入编辑模式

输入以下指令以添加变量

export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

编辑完成 按Esc退出 然后 :wq保存并退出

然后运行一下指令 激活环境 并查看版本

source ~/.bashrc
nvcc -V

但我做完了这些，却还是找不到GPU 

然后我看了下对应的版本 （见开头）

pip list 里版本太高 然后我直接 去这个网址 https://download.pytorch.org/whl/torch_stable.html 找到了对应的whl文件，下载 然后pip install xxxx.whl --force-reinstall 

直接强制重新安装了pip的 torch 然后版本对应了 就找到了

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！