ubuntu系统安装到pytorch安装,anaconda安装

ubuntu系统安装

  • 下载ubuntu18.04LTS,使用软碟通制作u盘系统盘
  • 安装ubuntu

显卡驱动安装

软件与更新那里,选择附件驱动,选择nvidia专用驱动

anaconda安装

  • anaconda下载到/home/dc/下载
  • 打开终端:
cd /home/dc/下载/
bash Anaconda3-2020.11-Linux-x86_64.sh

将终端中默认的base去掉:
conda config --set auto_activate_base false

cuda和cudnn安装

cuda官网扎到对应版本,安装,添加path,参考

conda使用

查看已有环境

conda info --envs

新建、激活、删除虚拟环境

conda create -n py1.9_cu10 python=3.7
conda activate py1.9_cu10
conda deactivate

安装pytorch-GPU版本

  • 参照官网命令
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

解决conda安装pytorch-gpu比较慢的问题

参考
具体如下:

  • 添加清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
  • 添加conda第三方源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  • 运行pytorch安装命令,注意这里去掉 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2

测试pytorch gpu版本是否安装成功

参考

import torch   # 能否调用pytorch库

print(torch.cuda.current_device())   # 输出当前设备(我只有一个GPU为0)
print(torch.cuda.device(0))   # 
print(torch.cuda.device_count())  # 输出含有的GPU数目
print(torch.cuda.get_device_name(0))  # 输出GPU名称 --比如1080Ti
x = torch.rand(5, 3)
print(x)  # 输出一个5 x 3 的tenor(张量)

pycharm安装

-下载pycharm到 /home/dc/下载/

  • 解压pycharm到/home/dc/下载/
  • 进入bin文件夹下安装
cd /home/dc/下载/pycharm-community-2021.2.2/bin
 sh ./pycharm.sh

创建pycharm快捷图标

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

将以下内容粘贴到文件中:

[Desktop Entry]
Version=1.0
Type=Application
Name=Pycharm
Icon=/home/dc/下载/pycharm-community-2021.2.2/bin/pycharm.png
Exec=sh /home/dc/下载/pycharm-community-2021.2.2/bin/pycharm.sh
MimeType=application/x-py;
Name[en_US]=pycharm

完成pycharm安装

在pycharm中添加anaconda虚拟环境

  • 新建项目
  • settings,选择conda,选择已经存在的虚拟环境py1.9_cu10

你可能感兴趣的:(ubuntu系统安装到pytorch安装,anaconda安装)