Ubuntu16.04安装了cuda后重启出现登录界面循环、分辨率降低的现象,可能是驱动出问题了以及版本号没对上。
解决方案:
第一步. 安装合适版本的驱动
第二步. 安装合适版本的cuda
第三步. 安装相应版本的pytorch
二、安装相应版本的pytorch
(一)卸载cpu版本的pytorch1.9.1
- 进入虚拟环境
- 卸载pytorch
(1)如果是用conda命令安装的,卸载用这个命令
conda uninstall pytorch
(2)如果是用pip命令安装的,卸载用这个命令
pip uninstall pytorch
- 输入卸载命令,输入Y
(二)安装GPU 版本的pytorch1.9.0
- 到官网查找合适版本的pytorch
https://pytorch.org/get-started/previous-versions/
- 选择合适的安装命令进行复制
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
- 到终端运行安装命令,成功后出现下图
- 查看pytorch版本
- 查看cuda的版本
- 验证是否可以调用cuda