Ubuntu20.04安装cuda cudnn pytorch pycharm记录

Ubuntu20.04安装cuda cudnn pytorch pycharm记录Ubuntu20.04安装cudacudnnpytorchpycharm记录0.安装NVIDIA驱动1.安装cuda(1)查看pytorch支持的cuda版本。(2)下载cuda安装包并安装2.安装cudnn3.安装Anaconda(略)4.conda换源、建立环境、pip换源(1)conda换源(2)建立conda环境(3)pip换源5.在环境中安装pytorch6.安装pycharm记录时间:2021年1月31日版本:Ubuntu20.04、cuda11.0、cudnn对应的版本、pytorc

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全家桶1年46,售后保障稳定

记录时间:2021年1月31日

版本:Ubuntu20.04、cuda11.0、cudnn对应的版本、pytorch对应的版本。我的电脑安装win10+Ubuntu20.04双系统,中途会重启进入windows系统进行一些下载。

0.安装NVIDIA驱动

Ubuntu20.04系统安装完成后,已有NVIDIA驱动,窗口输入nvidia-smi后显示驱动的信息。因此无需另外下载、安装驱动。此处显示驱动的版本,和最高支持的CUDA版本。“CUDA Version:11.2”指的是最高支持11.2版本的cuda,而后续不一定要安装11.2版本的cuda,也可以安装低于11.2的版本。

在这里插入图片描述

1.安装cuda

我们要配置pytorch的环境,首先要看pytorch支持哪些版本的cuda,而不能盲目选择。

(1)查看pytorch支持的cuda版本。

进入pytorch官网https://pytorch.org/,查看pytorch支持的cuda版本。

在这里插入图片描述
cuda 10.2针对的系统更偏向于Ubuntu18及以下的,我的系统是Ubuntu20.04,因此选择安装cuda11.0。

(2)下载cuda安装包并安装

进入cuda版本选择页面https://developer.nvidia.com/cuda-toolkit-archivE选择对应的版本;或直接再百度搜索cuda11进入指定版本的下载页面。

在这里插入图片描述
选择系统“Linux”,“x86_64”,“Ubuntu”,“20.04”。安装包的类型选择”runfile(local)”(因为其他类型的安装需要较多零散的安装包,runfile只有一个安装包,更便于下载)。wget会从官方网站下载,速度非常慢甚至会下载失败,因此我复制“wget”后面的网址,到windows系统中用迅雷下载此文件,大小上G。

[注:]如果迅雷下载的文件只有几个B大小,那肯定下载错了,需要重新复制链接进行下载。如果迅雷下载太慢甚至多次下载失败,则开会员可以解决。不同于去年我装系统用的更低版本的cuda,无需迅雷,直接官网下载速度也很快。要根据实际情况选择合适的下载方式。

在windows系统用迅雷下载好后,将.run安装包拷到Ubuntu系统中某个方便找到的文件夹下,在终端进入此文件夹,执行“sudo sh cuda_11.0.2_450.51.05_linux.run”,进入安装界面。安装选项除了450版本的driver不选(这是NVIDIA驱动,系统已经有了更高版本的,系统中NVIDIA驱动版本若是大于450.51.05,则不选;若小于此版本,则要选),其他都选。其他选项默认,很快就完成安装。

最后,添加环境变量。

sudo vim /etc/profile
#在最后添加两行
export PATH=/usr/local/cuda-11.0/bin${ 
   PATH:+:${ 
   PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${ 
   LD_LIBRARY_PATH:+:${ 
   LD_LIBRARY_PATH}}
#保存并退出
:wq 回车
#使它们生效
source /etc/profile 

Jetbrains全家桶1年46,售后保障稳定

在终端输入nvcc -V,若出现下图所示的结果,则说明安装成功、添加环境变量成功

在这里插入图片描述

验证cuda是否正常工作:依次在窗口中输入以下命令。输入最后一个命令,如果显示Result = Pass,则安装成功。

cuda-install-samples-11.0.sh ~  
cd ~/NVIDIA_CUDA-11.0_Samples 
make #需等待一段时间
./1_Utilities/deviceQuery/deviceQuery 

2.安装cudnn

进入cudnn官网https://developer.nvidia.com/cudnn登录后在Download的页面选择需要的版本。或直接进入版本选择页面https://developer.nvidia.com/cudnn。我选择下图的版本(为什么,因为其他版本我电脑上迅雷下载总是出错,这个不出错,所以选它)。

在这里插入图片描述
选择对应的安装包。我选择第一个(因为其他形式的安装包有好几个,就第一个安装包最简洁,所以选它)。同样,在windows系统中右键选择“迅雷下载”,下载此文件。

在这里插入图片描述
下载完成后,将文件拷到Ubuntu系统中。按照官方文档https://docs.nvidia.com/deeplearning/sdk/cudnn-install/的说明进行下载。我们下载了tar file,因此,参考官方文档的这一部分。

在这里插入图片描述

3.安装Anaconda(略)

这一部分网上有较多较好的教程,在此不赘述。建议从清华源等镜像源下载安装包。

4.conda换源、建立环境、pip换源

pytorch的安装会涉及到一些模块的下载,如果不换成国内的镜像源,往往导致下载失败。

(1)conda换源

在命令行输入命令 gedit ~/.condarc,去除-default,并添加以下内容:
[注:]我用这些源安装成功了,但还不完善,后续有更优质的源再继续添加。

auto_activate_base: false
show_channel_urls: true
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/

(2)建立conda环境

后面所有的模块都安装在集成环境中,环境也将作为pycharm中的interpreter。建立名为pytorch,包含python3.6的环境

conda create -n pytorch python=3.6

激活环境

conda activate pytorch

关闭环境

conda deactivate

(3)pip换源

pip包含在建立的conda环境中,先进入名为pytorch的环境,然后将pip换源

conda activate pytorch
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

5.在环境中安装pytorch

进入名为pytorch的环境(环境的名字可以任意选取)

conda activate pytorch

回到pytorch的官网,查看安装的方法。去除末尾的 -c pytorch,只输入“conda install pytorch torchvision torchaudio cudatoolkit=11.0”,开始安装。

在这里插入图片描述
安装完成后,在pytorch环境的python中import torch,如果不报错,则安装成功.

在这里插入图片描述

6.安装pycharm

进入pycharm官网https://www.jetbrains.com/pycharm/download/#section=linux,选择社区版(不会过期,不用激活码)

安装包下载后,解压,在终端打开解压后的文件夹。

cd Downloads/pycharm-版本号/bin
sh ./pycharm.sh

如果弹出窗口,默认选择就行,这时已经可以进入pycharm软件界面。

cd /usr/share/applications
sudo gedit pycharm.desktop

#输入以下内容
[Desktop Entry]
Version=1.0
Type=Application
Name=Pycharm
Icon=/home/XXXXXXXX/Downloads/pycharm-版本号/bin/pycharm.png
Exec=sh /home/XXXXXXX/Downloads/pycharm-版本号/bin/pycharm.sh
MimeType=application/x-py;
Name[en_US]=pycharm

这时就能在软件列表里找到pycharm软件。右键“add to favorates”,就能把它添加到任务栏。

如果在任务栏仍不能启动pycharm,则先从刚刚下载的安装包中启动pycharm.sh。打开pycharm后,选择Tools—-Create desktop entry,这时就能在软件中看到pycharm的图标,点击右键,“add to favorates”就能添加到桌面任务栏了。
打开搜索软件

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/210115.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)
blank

相关推荐

  • arrayqueue源码_thinkphp源码分析

    arrayqueue源码_thinkphp源码分析愉快地聊一聊ArrayDeque的特点吧~(以下都是基于jdk1.8)一棵树ArrayDeque的继承树如下图:基本特点(1)双端队列,可从两端添加、删除元素。作为队列使用时,性能优于LinkedList。作为栈使用时,性能优于Stack。(2)底层使用可变数组Object[]elements,数组容量按需增长(3)不能存储null(4)支持双向迭代器遍历(5)线程不安全…

  • 存储结构二叉树

    存储结构二叉树

  • PE框架学习之道:PE框架——style的配置

    PE框架学习之道:PE框架——style的配置

  • js正则 手机号_正则表达式 手机号

    js正则 手机号_正则表达式 手机号RegExp对象正则表达式是描述字符模式的对象。正则表达式用于对字符串模式匹配及检索替换,是对字符串执行模式匹配的强大工具。//匹配手机号//varres=/^[1][3,4,5,6.7,8,9][0-9]{9}$///letnum=18045216903//console.log(res.test(num))//匹配身份证号码//varres=/(^\d{15}$)|(^\d{18}$)|(^\d{17}(\d|X|x)$)///varn=’5203

  • CTK框架介绍

    CTK框架介绍转(http://blog.csdn.net/xinqidian2015/article/details/50537325)CTK插件框架可以简单的描述为C++的动态组件系统DesignCTK插件框架的设计有很大的灵感来自OSGi并且使得应用程序由许多不同的组件组合成一个可扩展模型。这个模型允许通过那些组件间共享对象的服务通信。框架的分层模型被展示在图片1中包括:P

  • PCL 平面点云B样条曲线拟合

    PCL 平面点云B样条曲线拟合平面点云B样条曲线拟合

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号