本文共 875 字，大约阅读时间需要 2 分钟。

我的配置

Ubuntu 16.04

准备工作

参考这篇文章：

1. 添加Anaconda的TUNA镜像（直接安装就会被墙）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

2. 设置搜索时显示通道地址（这一项应该不是必须的）

conda config --set show_channel_urls yes

3. 执行过conda config指令后，就会生成~/.condarc文件，修改这个文件，删除其中的default行

sudo gedit .condarc

修改后的.condarc文件长这样：

安装

PyTorch安装指令经常更改，最好进入查看

我装的时候是这样的：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

测试

1. 进入Python，运行下面一段程序

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

输出的结果类似这样，表明PyTorch已经正确安装

tensor([[0.3380, 0.3845, 0.3217],

2. 运行下面的程序：

import torchtorch.cuda.is_available()

检查PyTorch能否访问并启用GPU驱动和CUDA

转载地址：https://blog.csdn.net/weixin_53381817/article/details/111366626 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！