查看 CUDA \cudnn版本,使用conda/pip安装pytorch教程(mac/windos/linux) 和卸载

2019.9.18

安装时候仔细看安装说明,cuda、cudnn、系统的版本需要对应。

2019.01.01 更新

可以直接去官网看安装指令,里面也有一些旧版本的指令入口:https://pytorch.org/

2019.02.16 更新

网速问题的话,安装指令示意:https://blog.csdn.net/m0_37644085/article/details/87446922

#阿里云
eg:安装:
pip install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com torch
eg:更新:
pip install --upgrade -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com torch
#清华
eg:更新
pip install --upgrade -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host https://pypi.tuna.tsinghua.edu.cn/ torch  
eg:安装:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host https://pypi.tuna.tsinghua.edu.cn/ torch  

查看 CUDA \cudnn版本,使用conda/pip安装pytorch教程(mac/windos/linux) 和卸载_第1张图片

 

 一、查看版本

先查看 CUDA 版本:

cat /usr/local/cuda/version.txt

查看 CUDNN 版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

查看 CUDA \cudnn版本,使用conda/pip安装pytorch教程(mac/windos/linux) 和卸载_第2张图片

二、测试

在终端中按以下顺序运行命令:

python

在打开的python命令行中运行

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

不报错的话即代表安装成功。最好再测个cuda能不能用,返回True。

三、卸载Pytorch

1、使用conda卸载Pytorch

conda uninstall pytorch 
conda uninstall libtorch

2、使用pip卸载Pytorch

pip uninstall torch
 

你可能感兴趣的:(遇到的问题,LINUX)