Conda的PyTorch安装

一、前言

今天把代码部署到了一个新服务器上，想要运行却发现代码运行极其缓慢，推测是代码运行在了CPU上。

于是尝试执行了如下语句

import torch
print(torch.cuda.available())

打印的结果为False，说明此时的cuda不可用，推测正确，代码确实是在CPU上跑的。

于是执行nvidia-smi查看显卡，发现显卡cuda版本是11.3，而执行conda list，发现torch版本为1.7.0，上torch官网查询后发现，torch的1.7.0版本最高支持11.1版本的cuda。

破案了，于是进行重装。

二、配置镜像源

在经历了多次的龟速直接安装后，还是决定使用镜像源来安装。

这里我们使用清华镜像源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

添加成功后，我们即可使用清华源进行安装。

三、安装PyTorch

最后一步，需要前往PyTorch官网，根据自己的配置进行选择

如图，我安装的是Linux上1.11.0版本的PyTorch和11.3的CUDA，选择完毕后，官网会直接给出命令

但注意：命令需要修改，conda install的-c参数代表指定channel，由于我们已经配置了清华源，故可以不用指定channel直接执行命令。

命令如下：

conda install pytorch torchvision torchaudio cudatoolkit=11.3

执行完后即可安装成功！

Conda的PyTorch安装相关推荐

笔记 | PyTorch安装及入门教程
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达本文内容概述如何安装PyTorch以及PyTorch的一些简单操作 ...
Tesla M40 下Ubuntu anaconda pycharm pytorch安装
Tesla M40 训练机组装参考:http://t.csdn.cn/7Fx13,写的很详细显卡:Tesla M40 24GB (2张) 显卡驱动版本(推荐):470.57.02 cuda版本:11 ...
AI 图片截取、ffmpeg使用及安装, anaconda环境，图片标注(labelme)，模型训练(yolov5)，CUDA+Pytorch安装及版本相关问题
AI 图片截取(ffmpeg), anaconda环境,图片标注(labelme),模型训练(yolov5),CUDA+Pytorch安装及版本相关问题一.截取有效图片录制RTSP视频脚本 #!/ ...
PyTorch安装及配置
PyTorch 环境配置及安装 1. 安装 anaconda 2. 为 PyTorch 创建单独的环境,便于管理. 打开 Anaconda Prompt 为 PyTorch 创建单独的环境 pytho ...
conda清华源安装pytorch依旧慢踩坑
之前也安装过很多次,但这次发现换了清华源还是很慢,后来得知. 官网上的命令是这样的 conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolki ...
Ubuntu安装显卡驱动，配置conda环境，安装pytorch
安装显卡驱动这里选择自动安装,先更新升级 sudo apt-get update sudo apt-get upgrade 自动选择驱动安装 sudo ubuntu-drivers autoinst ...
pytorch安装笔记——Pytorch在conda+CUDA10.2环境安装task01
Pytorch在conda+CUDA10.2环境安装前言由于之前电脑里安装的CUDA是10.2版本的,paddlepaddle什么的也装的是CUDA10.2的版本,不想再去重装CUDA. 如图:官 ...
pytorch安装换源ubuntu_ubuntu 安装pytorch问题
由于之前pytorch安装在base环境中,现在弄了个新环境py37env,想在上面安装pytorch 但是这次安装出现了一点问题报错说 Found conflicts! Looking for i ...
python --- 使用conda配置pytorch
使用Conda配置PyTorch 1. 添加channels 下载地址 $ conda config --add channels https://mirrors.tuna.tsinghua.edu. ...
清华源离线安装pytorch_【Pytorch】pytorch安装的时候下载不动
第一步,在对应环境中配置清华源 : conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/fre ...

Conda的PyTorch安装

Conda的PyTorch安装

一、前言

二、配置镜像源

三、安装PyTorch

Conda的PyTorch安装相关推荐

最新文章

热门文章