Ubuntu18.04——安装Anaconda和python3.6.9版本的pytorch-gpu

文章目录

• ​​一、安装Anaconda​​

• ​​1.下载任意版本的conda安装脚本​​
• ​​2.安装​​
• ​​3.测试​​
• ​​4.换源​​
• ​​卸载（如有需要）​​

• ​​安装pytorch​​

• ​​创建conda虚拟环境​​

• ​​卸载虚拟环境（如有需要）​​

• ​​查看cuda版本​​
• ​​选择pytorch对应版本​​

• ​​安装​​

• ​​测试​​

• ​​如果出现False：​​
• ​​1.检查安装的pytorch是否为GPU版本​​
• ​​2.卸载cpu版本​​

一、安装Anaconda

需要说明的是，安装anaconda的对应python版本不影响我们在使用过程中创建其他python版本的环境，如我下载的是：python3.8版本的conda，那我在使用过程中依然可以通过指定python3.6版本的方式创建环境，conda会自动下载python3.6版本的所有基本包，你也可以自己另外安装其他所需要的任何库。

1.下载任意版本的conda安装脚本

• ​​所有版本官网下载地址​​
• ​​python和conda对应版本查询文档​​

2.安装

curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
sh Miniconda3-latest-Linux-x86_64.sh

之后长安回车和全输入yes即可。
我的安装路径设置为：：

/home/pc/Softwares/anaconda/miniconda3

3.测试

source ~/.bashrc
python

如果出现，则表示安装成功：

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/pytorch/
conda config --set show_channel_urls yes

卸载（如有需要）

• ​​ubuntu18.04 卸载Anaconda3​​

安装pytorch

创建conda虚拟环境

conda create -n pytorch python=3.6.9
conda activate pytorch

卸载虚拟环境（如有需要）

conda deactivate # 退出虚拟环境
conda remove -n pytorch --all

查看cuda版本

nvcc -V

选择pytorch对应版本

• ​​官网地址​​

安装

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

测试

进入python环境：

python

输入：

import torch
import torchvision
print(torch.cuda.is_available())

如果出现False：

1.检查安装的pytorch是否为GPU版本

果然后缀为cpu：

2.卸载cpu版本

conda uninstall pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

在网上发现pytorch1.4.0版本没有cuda10.2对应版本的whl，所以现在在下载时要指定版本：（​​官网版本查询处​​）

conda install pytorch=1.5 torchvision=0.6 cudatoolkit=10.2 -c pytorch

安装前检查：

测试：

参考文章：

• ​​Anaconda与Python版本对应关系 [官方最新]​​
• ​​Ubuntu18.04 安装 Conda环境​​
• ​​完美解决torch.cuda.is_available()一直返回False的玄学方法​​