Pytorch安装(GPU版本)
第一步
- 进入anaconda官网(也可以是清华镜像源)进行对应系统下载(本教程基于Ubuntu20.04下运行)
- 进入清华镜像源,点击获取下载链接
- 点击应用软件
- 点击Conda
- 进行下载对应安装包进行解压
- 进入
sudo gedit ~/.bashrc
添加export PATH=/root/anaconda3/bin:$PATH
随后source ~/.bashrc
生效
第二步
- 使用conda命令创建虚拟环境(本教程基于python3.7)
conda create -n pytorch3.7 python=3.7
- 进入虚拟环境中
source activate pytorch3.7
- 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
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本