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 然后版本对应了 就找到了