文章目录 前提 安装显卡驱动 查看显卡驱动型号 安装Anaconda,创建虚拟环境 安装 PyTorch1.1.0 版本 安装torvision 安装six,PIL包 验证环境是否安装到位 结束语 前提 安装好Ubuntu16.04系统,更换软件源到国内的镜像地址,有nvidia的GPU。 如图所示: 安装显卡驱动 最简单的方法就是在“软件和更新”中的“附加驱动”安装。 查看显卡驱动型号 确定各自的Pytorch、cuda与torchvison版本。 终端输入命令:nvidia-smi 查表可知对应包版本各为: CUDA9.0,pytroch1.1.0,torchvision0.3.0