Pytorch安装以及环境配置

1. 首先查看系统的cuda版本的配置 

      nvcc -V

      检查cuda 的编译版本是 nvidia-smi,但是一般使用的是nvcc -V

  2、尤其要注意pytorch 、Python 、cuda 版本的对应问题

查看Python版本 python --version 

• 查看cuda版本：cat /usr/local/cuda/version.txt
•  torch、torchvision、cuda 、python对应版本匹配

• 

 3、添加镜像源

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

4、开始安装

创建环境 conda create -n pytorch python=3.7

5、激活环境

conda activate pytorch

6、进入官网，选择与运行cuda 版本一致的pytorch安装

PyTorch

7、检查torch是否可用

# 验证安装是否成功
# 使用python运行
import torch
print(torch.__version__)
print(torch.cuda.is_available())

检查PyTorch版本

1 2 3 4

torch.version # PyTorch version torch.version.cuda # Corresponding CUDA version torch.backends.cudnn.version() # Corresponding cuDNN version torch.cuda.get_device_name(0) # GPU type

 

7.1 验证torch能不能正常使用，正常打印则说明安装成功

import torch
x = torch.rand(5, 3)
print(x)

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

8、其他命令

conda env list 
conda info --envs
#列出所有环境
conda activate ee #激活环境
conda deactivate #退出环境
conda env remove -n py3 #删除环境
conda create -n pytorch python=3.6 #创建环境
conda activate pytorch
#安装 cpu 版本的 PyTorch
conda install -c pytorch pytorch-nightly-cpu
#安装fastai
conda install -c fastai fastai
#更新conda
conda update conda
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
#安装gpu pytorch
# 1.conda新建环境 
# 2.切换到新建环境中 
# 3.安装ipykernel conda install ipykernel 
# 在新建环境新建kernel（kernel名可与环境名一致） 
python -m ipykernel install --user --name <环境名> --display-name <在jupyter中显示的kernel名称>
python -m ipykernel install --user --name pytorch --display-name Pytorch
jupyter kernelspec list # 查看jupyter的kernel列表 
jupyter kernelspec remove <kernel_name> # 删除指定kernel

 链接：

https://www.jb51.net/article/207745.htm 

​​​​​​PyTorch(CPU版本及GPU版本)的安装及环境配置_L_bloomer的博客-CSDN博客_pytorchgpu版本

安装cuda 10.1和pytorch - 走看看