Linux环境下GPU版本的pytorch安装

更新时间：2021年05月16日 10:08:34 作者：五岳凌峰

使用默认的源地址下载速度很慢，所以一般都是使用国内源，今天花了点时间配置安装，所以记录一下，需要的朋友们下面随着小编来一起学习学习吧

服务器环境：

Ubuntu 16.04.7

显卡：2080

cuda:10.1

注：若服务器有管理员账户和个人账户，最好在个人账户下重新安装anaconda，否则安装pytorch过程中可能有些库安装失败，由于权限问题，不能删除这些失败的库重新安装。在个人账户下就不存在权限问题。

一添加镜像源

目的：使用默认的源地址下载速度很慢，会出现超时，导致某些第三方库只下载了部分，不完整，最终失败。

首先查看当前镜像源

cat ~/.condarc

或者

conda config --show channels

或者

conda info

然后添加清华镜像源。参考官网https://mirror.tuna.tsinghua.edu.cn/help/anaconda/和https://www.jb51.net/article/212360.htm

先尝试添加1个镜像源(可以尝试只添加第一个main就够了,路径下包括常用的第三方库如python等，不够的话按需要再添之后几个)。个人最终添加的是如下4个

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
conda config --add channels https://mirrors.t

然后查看当前镜像源，发现多了刚添加的镜像源网址

注：如果需要增加更多的镜像源，仿照上面命令继续添加即可。若需求更换其他镜像源，可以把刚才的删除，删除刚添加的镜像源命令：

conda config --remove channels https://mirrors.

添加镜像源结束后，输入命令

conda config --set show_channel_urls yes

conda config --set show_channel_urls yes的意思是从channel中安装包时显示channel的url，这样就可以知道包的安装来源了。

输入如下命令，清除索引缓存，保证用的是镜像站提供的索引

conda clean -i

二创建虚拟环境

conda create -n myenv python=3.5.2

注：移除环境命令，conda remove -n 环境名 --all

注：报错

错误1：若5分钟以上后创建失败，报错

请参考博客https://www.jb51.net/article/212362.htm

错误2：Segmentation fault (core dumped)

原因：在未加清华镜像源的时候，尝试创建虚拟环境，由于国外镜像源超时导致失败，有些包只下载了部分，产生了历史缓存。需要清空。输入如下命令，按提示操作

conda clean –a

创建过程及成功后，有如下页面信息

三安装pytorch

从base切换到上面创建的新虚拟环境，

conda activate pytorch_gpu_lsq

查看已安装的cuda版本，查看官网https://pytorch.org/，根据cuda版本和操作系统类型/版本等信息，选择安装哪个版本的pytorch。服务器cuda 10.1，所以安装命令如下

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

需要一些时间，等待即可。

到此这篇关于Linux环境下GPU版本的pytorch安装的文章就介绍到这了,更多相关Pytorch GPU安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

python PyAutoGUI 模拟鼠标键盘操作和截屏功能
一款跨平台/无依赖的自动化测试工具，目测只能控制鼠标/键盘/获取屏幕尺寸/弹出消息框/截屏。这篇文章主要介绍了python PyAutoGUI 模拟鼠标键盘操作和截屏功能,需要的朋友可以参考下
2019-08-08
Python中定时任务框架APScheduler的快速入门指南
APScheduler是基于Quartz的一个Python定时任务框架，实现了Quartz的所有功能，使用起来十分方便。下面这篇文章主要跟大家介绍了Python中定时任务框架APScheduler的快速入门指南，需要的朋友可以参考借鉴，下面来一起看看吧。
2017-07-07
基于Python轻松实现PDF转图片
PDF文件是我们在日常工作和学习中常用的文档格式之一,但你知道吗,你可以将PDF文件转换为图像,让文档变得更加生动有趣,下面我们就来看看具体的实现方法吧
2023-08-08
使用Python集合显著优化算法性能的实战案例
掌握 Python 中的 set 数据结构，是算法和数据结构的基本功，今天我们从一个实战案例出发，探讨如何利用Python集合显著优化算法性能，感兴趣的同学跟着小编一起来探讨吧
2023-06-06
python随机在一张图像上截取任意大小图片的方法
今天小编就为大家分享一篇python随机在一张图像上截取任意大小图片的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-01-01
scrapy爬虫实例分享
这篇文章主要介绍了scrapy爬虫实例分享，分享了四则Python爬虫实例，具有一定借鉴价值,需要的朋友可以参考下
2017-12-12
Python实现socket非阻塞通讯功能示例
这篇文章主要介绍了Python实现socket非阻塞通讯功能,结合实例形式分析了Python使用socket模块进行非阻塞通讯的原理、多线程及客户端、服务器端相关实现技巧,需要的朋友可以参考下
2019-11-11
python+Django实现防止SQL注入的办法
这篇文章主要介绍了python+Django实现防止SQL注入的办法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-10-10
Python编程使用Selenium模拟淘宝登录实现过程
这篇文章主要介绍了Python编程使用Selenium模拟淘宝登录的实现过程示例及解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步早日升职加薪
2021-10-10
在python中读取和写入CSV文件详情
这篇文章主要介绍了在python中读取和写入CSV文件详情，CSV即逗号分隔值，一种以逗号分隔按行存储的文本文件，所有的值都表现为字符串类型
2022-06-06