广告位联系
返回顶部
分享到

Linux环境下GPU版本的pytorch安装

python 来源:转载 作者:秩名 发布时间:2021-05-16 14:31:25 人浏览
摘要

服务器环境: Ubuntu 16.04.7 显卡:2080 cuda:10.1 注:若服务器有管理员账户和个人账户,最好在个人账户下重新安装anaconda,否则安装pytorch过程中可能有些库安装失败,由于权限问题,不能删除这些失败的库重新安装。在个人账户下就不存在权限问题。 一

服务器环境:

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分钟以上后创建失败,报错

错误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

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


版权声明 : 本文内容来源于互联网或用户自行发布贡献,该文观点仅代表原作者本人。本站仅提供信息存储空间服务和不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权, 违法违规的内容, 请发送邮件至2530232025#qq.cn(#换@)举报,一经查实,本站将立刻删除。
原文链接 : https://blog.csdn.net/u012911202/article/details/116764278
相关文章
  • Python Django教程之实现新闻应用程序

    Python Django教程之实现新闻应用程序
    Django是一个用Python编写的高级框架,它允许我们创建服务器端Web应用程序。在本文中,我们将了解如何使用Django创建新闻应用程序。 我们将
  • 书写Python代码的一种更优雅方式(推荐!)

    书写Python代码的一种更优雅方式(推荐!)
    一些比较熟悉pandas的读者朋友应该经常会使用query()、eval()、pipe()、assign()等pandas的常用方法,书写可读性很高的「链式」数据分析处理代码
  • Python灰度变换中伽马变换分析实现

    Python灰度变换中伽马变换分析实现
    1. 介绍 伽马变换主要目的是对比度拉伸,将图像灰度较低的部分进行修正 伽马变换针对的是对单个像素点的变换,也就是点对点的映射 形
  • 使用OpenCV实现迷宫解密的全过程

    使用OpenCV实现迷宫解密的全过程
    一、你能自己走出迷宫吗? 如下图所示,可以看到是一张较为复杂的迷宫图,相信也有人尝试过自己一点一点的找出口,但我们肉眼来解谜
  • Python中的数据精度问题的介绍

    Python中的数据精度问题的介绍
    一、python运算时精度问题 1.运行时精度问题 在Python中(其他语言中也存在这个问题,这是计算机采用二进制导致的),有时候由于二进制和
  • Python随机值生成的常用方法

    Python随机值生成的常用方法
    一、随机整数 1.包含上下限:[a, b] 1 2 3 4 import random #1、随机整数:包含上下限:[a, b] for i in range(10): print(random.randint(0,5),end= | ) 查看运行结
  • Python字典高级用法深入分析讲解
    一、 collections 中 defaultdict 的使用 1.字典的键映射多个值 将下面的列表转成字典 l = [(a,2),(b,3),(a,1),(b,4),(a,3),(a,1),(b,3)] 一个字典就是一个键对
  • Python浅析多态与鸭子类型使用实例
    什么多态:同一事物有多种形态 为何要有多态=》多态会带来什么样的特性,多态性 多态性指的是可以在不考虑对象具体类型的情况下而直
  • Python字典高级用法深入分析介绍
    一、 collections 中 defaultdict 的使用 1.字典的键映射多个值 将下面的列表转成字典 l = [(a,2),(b,3),(a,1),(b,4),(a,3),(a,1),(b,3)] 一个字典就是一个键对
  • Python淘宝或京东等秒杀抢购脚本实现(秒杀脚本

    Python淘宝或京东等秒杀抢购脚本实现(秒杀脚本
    我们的目标是秒杀淘宝或京东等的订单,这里面有几个关键点,首先需要登录淘宝或京东,其次你需要准备好订单,最后要在指定时间快速
  • 本站所有内容来源于互联网或用户自行发布,本站仅提供信息存储空间服务,不拥有版权,不承担法律责任。如有侵犯您的权益,请您联系站长处理!
  • Copyright © 2017-2022 F11.CN All Rights Reserved. F11站长开发者网 版权所有 | 苏ICP备2022031554号-1 | 51LA统计