ubuntu18.04配置pytorch环境
- 安装ubuntu18.04
- 安装显卡驱动
- 安装anaconda(此处安装python3.6对应版本)
- 安装cuda、cudnn(需要查版本)
- 安装pytorch、torchvision包
其中在安装pytorch和torchvision包时,官网安装太慢,并且总是中断,于是去官网下载好对应版本安装包,我选的是Python3.6版本,cuda10.1版本。
然后进入下载目录打开终端,离线使用pip进行安装(这里有一个插曲是我安装时使用sudo pip 一直安装不成功,注意了提示发现,sudo并非当前用户,于是不用sudo正常安装pytorch和torchvision包一次成功)