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