pytorch 、torchvision、cuda 、cudatoolkit在Ubuntu下版本问题+python版本安装问题

pytorch 、torchvision、cuda 、cudatoolkit在Ubuntu下

查看pytorch版本

import torch

import torch
torch.__version__

torch.__version__ （两边个两条下横线）

查看torchvision版本

import torchvision
torchvision.__version__  

查看torchvision版本

nvcc -V

查看cuda版本：

nvcc -V  
#对应CUDA Runtime API

#或

nvidia-smi
#对应CUDA Driver API

pytorch 、torchvision、cuda 、cudatoolkit版本对照链接
上面3个查了其中一个或者两个再来这里面对照下载安装即可。因为很多时候版本不对应容易出问题。

官网：

Previous PyTorch Versions | PyTorch

或者查看此链接Linux下cuda10.0安装Pytorch和Torchvision——啥版本都能装_墨理学AI的技术博客_51CTO博客

补充：若python版本安装错误，可删除当前版本再重新安装

（1）删除

conda remove python=版本号 -y

（2）安装新版本

conda install python=版本号 -y