ubuntu20.04系统 conda安装pytorch-GPU版本

编程研究坊

已于 2023-02-15 12:48:04 修改

阅读量1k

点赞数

分类专栏： python 人工智能算法文章标签： pytorch conda 深度学习

于 2023-02-10 17:05:17 首次发布

本文链接：https://blog.csdn.net/li1873997/article/details/128973689

版权

人工智能同时被 3 个专栏收录

74 篇文章

订阅专栏

算法

67 篇文章

订阅专栏

python

49 篇文章

订阅专栏

本文指导用户通过conda在已安装cuda11.2的环境下，安装pytorch的GPU版本1.12.1，torchvision0.13.1和torchaudio0.12.1，指定cuda工具包为11.3，并提示需移除-cpytorch参数以使用清华源。安装后，通过检查cuda版本及cuDNN的兼容性验证安装成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

去官网查询conda安装pytorch-GPU版本命令：https://pytorch.org/get-started/previous-versions/

本文选用cuda11.3(本机已经装过cuda11.2)

一定要去掉 -c pytorch！！！！这样才能使用清华源

onda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3

-c后面带的是镜像地址如：

conda install torchaudio==0.9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

检测是否安装成功

import torch
import torchvision
torch.cuda.is_available() #True即安装成功

安装后查询发现cuda还是11.2

测试CUDA和cuDNN

# CUDA TEST

import torch

x = torch.Tensor([1.0])

xx = x.cuda()

print(xx)

# cuDNN test

from torch.backends import cudnn

print(cudnn.is_acceptable(xx))