在Ubuntu系统中搭建gpu版pytorch环境
一、配置信息
Ubuntu18.04,操作系统64位;
显卡:GeForce GTX 950M;
显卡驱动版本:450.57
Anaconda版本:Anaconda3 2020.07
CUDA:10.0.130
cuDNN:7.6.0
Pytorch:1.2.0
二、搭建pytorch-gpu环境
1、重装显卡驱动
安装好的Ubuntu18.04自带了显卡驱动,但不能在终端使用nvidia-smi命令查看,搭建环境前要先重装,否则会出现大麻烦,之前由于没注意这个问题导致重装了两次ubuntu。
打开终端,检测N卡和推荐驱动输入:
ubuntu-drivers devices
下图是我的显卡和推荐驱动信息,可以看到显卡型号和支持的驱动版本,我就安装推荐的nvidia-driver-450版。
选择安装所推荐的驱动,输入:
sudo ubuntu-drivers autoinstall
安装完成后注意一定要重启电脑,打开设置中的详细信息一栏,可以在图形中看到显卡型号就代表安装驱动成功了。
2、安装Anaconda
到官网去下载Anaconda,地址:https://www.anacond