Anaconda3+Pycharm+Pytorch+GPU

一、安装显卡驱动

1.禁用nouveau

终端输入下面两行命令：

sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"

sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"

查看否成功禁用了开源驱动：

cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf

显示如下则成功：

blacklist nouveau

options nouveau modeset=0

2.安装驱动

打开附加驱动，选择推荐的专有驱动，选择应用更改

二、安装Anaconda

1.下载Anaconda

官网：https://www.anaconda.com/

清华镜像：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

清华镜像选择Anaconda3-2020.11-Linux-x86_64.sh

2.安装Anaconda

先进入下载文件所在目录，再执行如下命令：

bash Anaconda3-2020.11-Linux-x86_64.sh

接下来按照提示操作：直接回车

继续回车

输入“yes”，回车；

默认安装路径在“/home/username/Anaconda”，直接回车安装在默认路径；或者输入其他安装路径：

Do you wish the installer to initialize Anaconda3 by running conda init?  输入yes，回车：

3.重新启动终端：出现base；

查看anaconda相关的信息：

conda info

4.更换清华源以提高相关包的下载速度

终端输入：

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
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/pytorch/
conda config --set show_channel_urls yes

查看相关源：

conda config --show-sources

修改 .condarc文件，删除“-defaults”：

sudo gedit .condarc

三、安装Pytorch

1.创建虚拟环境

conda create -n pytorch pip python=3.7

2.进入虚拟环境

conda activate pytorch

3.安装Pytorch

进入官网，选择所需版本，复制下方给出的代码：

安装时删除“-c pytorch”，否则还是从官方源安装，速度慢

conda install pytorch torchvision torchaudio cudatoolkit=10.2

4.测试安装是否成功

如果没有报错，则环境配置成功。

python3
import torch
import torchvision

5.更换pip源（可选）

在用户目录下（/home/username）创建新的文件夹 .pip

sudo mkdir .pip

进入 .pip文件夹，创建pip.conf

cd  .pip      
sudo touch pip.conf

在pip.conf文件中写入：

sudo gedit ~/.pip/pip.conf

[global]
index-url=http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host=mirrors.aliyun.com

四、安装Pycharm

1.下载Pycharm

进入官网，选择Community版本。

2.安装Pycharm

第一，解压文件：右键-->提取到此处；

第二，将解压后的文件放入安装目录；

第三，终端进入pycharm的bin文件夹；

第四，执行以下命令，开始安装：

sh ./pycharm.sh

 

参考文章：

pip 源使用阿里云镜像加速

Linux系统下利用anaconda创建虚拟环境并安装pytorch0.4.1和torchvision

Ubuntu 18.04 安装 PyCharm