Pytorch安装(GPU版本)

第一步

  1. 进入anaconda官网(也可以是清华镜像源)进行对应系统下载(本教程基于Ubuntu20.04下运行)
  2. 进入清华镜像源,点击获取下载链接
  3. 点击应用软件
  4. 点击Conda
  5. 进行下载对应安装包进行解压
  6. 进入 sudo gedit ~/.bashrc 添加 export PATH=/root/anaconda3/bin:$PATH 随后 source ~/.bashrc 生效

第二步

  1. 使用conda命令创建虚拟环境(本教程基于python3.7)
    conda create -n pytorch3.7 python=3.7
  2. 进入虚拟环境中
    source activate pytorch3.7
  3. conda换清华镜像源
# 换源
#查看当前conda配置
conda config --show channels
#设置通道
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
#设置搜索是显示通道地址
conda config --set show_channel_urls yes
# 换回默认源
conda config --remove-key channels

## 在命令行中运行pytorch中给出的代码(记住去除-c pytorch)
# cuda117 !!!清华镜像中无cudatoolkit=11.7这个包,故运行结束后再次运行官方版本(conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia)
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 
# cuda113
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3
# 如果安装太慢 可尝试以下命令
conda config --show-sources
rm -rf .condarc

## 安装成功后,输入python
python
## 进入python编辑器后输入
>>> import torch
>>> torch.cuda.is_available()
## 如果输出True,则说明安装成功GPU版本,False是CPU版本

第三步

如果安装不成功,则进行一下操作

## 进入anaconda3的pkgs文件夹,输入
# cuda117 cuda113同理
wget https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.13.0-py3.7_cuda11.7_cudnn8.5.0_0.tar.bz2
wget https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/linux-64/torchaudio-0.13.0-py37_cu117.tar.bz2
wget https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/linux-64/torchvision-0.14.0-py37_cu117.tar.bz2

第四步

确保正在所创建的虚拟环境中(本文为pytorch3.7)

# cuda117 cuda113同理
conda install --use-local pytorch-1.13.0-py3.7_cuda11.7_cudnn8.5.0_0.tar.bz2
conda install --use-local torchvision-0.14.0-py37_cu117.tar.bz2
conda install --use-local torchaudio-0.13.0-py37_cu117.tar.bz2
## 安装成功后重复第三步第3小步,检测是否安装成功

opacus安装

# 1.1.1版本的opacus
## 指定清华源 pip install -r requirements.txt -i http://pypi.tuna.tsinghua.edu.cn/simple/
pip install opacus==1.1.1
posted @   WTALingLong  阅读(2224)  评论(1编辑  收藏  举报
相关博文:
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
点击右上角即可分享
微信分享提示
主题色彩