ubuntu18.04安装cuda+cudnn+anaconda+pytorch+pycharm

ubuntu18.04安装cuda+cudnn+anaconda+pytorch+pycharm,第1张

1.前期准备 1.1安装双系统

win11+ubuntu18.04双系统安装教程_Happy_Cabbage的博客-CSDN博客_ubuntu18.04双系统安装教程

1.2打开terminal
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential
1.3查看推荐的显卡版本
ubuntu-drivers devices

记住recommended驱动

打开软件和更新->附加驱动->选择推荐的那个(一般是带tested的那个)

2.安装cuda

一定要选择合适的cuda版本!!!我的是11.3.1

RTX3080+cuda问题_weixin_43751285的博客-CSDN博客_3080支持cuda版本

nvidia显卡和CUDA版本关系 - 简书

下载cuda

CUDA Toolkit Archive | NVIDIA Developer

安装

Ubuntu18.04安装cuda10.0_星星点灯@的博客-CSDN博客_ubuntu安装cuda10.0

3.安装cudnn

一定要选择与cuda版本对应的!!!我的是8.2.1

https://developer.nvidia.com/rdp/cudnn-download?spm=a2c4e.10696291.0.0.1df819a4HJWSTe

安装(采用方法2)

Ubuntu 20.04安装CUDA 11.0、cuDNN 8.0.5、PyTorch 1.7.0 - NobodySwordsmith - 博客园

4.安装anaconda

查看版本!!!我的是3-5.2.0(对应python3.6.5)

anaconda python 版本对应关系__沥川往事的博客-CSDN博客_anaconda python版本

下载

Index of / (anaconda.com)

安装 

在.sh文件下打开终端,输入

bash ./Anaconda3-5.3.0-Linux-x86_64.sh

yes Enter yes no

重启终端

source activate
5.安装pytorch

新建环境

conda create -n pytorch python=3.6

激活环境

conda activate pytorch

打开pytorch官网选择相应的版本,复制pip命令进行安装

PyTorch  

检查安装的包

pip list

检查是否安装成功

python
import torch

 没报错说明安装成功

torch.cuda.is_available()

显示为True,说明GPU可被pytorch使用

6.安装pycharm

Ubuntu 18.04 安装 PyCharm_梦dancing的博客-CSDN博客_ubuntu安装pycharm

下载community版本 

7.配置环境

create new project -> existing interpreter -> ... -> conda environment -> pytorch

python console :import torch ->torch.cuda.is_available()

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/718581.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-25
下一篇 2022-04-25

发表评论

登录后才能评论

评论列表(0条)

保存