1. 首页
  2. 编程语言
  3. Python
  4. Linux安装Pytorch1.8GPU(CUDA11.1)的实现

Linux安装Pytorch1.8GPU(CUDA11.1)的实现

上传者: 2022-02-12 16:49:07上传 PDF文件 628.96 KB 热度 9次

提示,如果想要保留之前的PyTorch1.6或1.7的环境,请不要卸载CUDA环境,可以通过Anaconda管理不同的环境,互不影响。但是需要注意你的在这里能够看到官方给的对应CUDA版本所需使用驱动版本。通过上表可以发现,如果要使用CUDA11.1,那么需要将显卡的驱动更新至455.23或以上。更新驱动卸载旧驱动我之前安装的是NVIDIA-440的版本,找到之前下载的安装程序,然后打开终端通过以下指令进行卸载:安装新驱动1)下载驱动,直接去NVIDIA官网下载:根据你的GPU型号以及操作信息选择对应的驱动,注意CUDA Toolkit11版的当前可选的只有11.0和11.2,而我们要装的是11.1所以选择11.2即可。2)关闭Xserver服务 显示结果:

用户评论