RTX3050ti安装pytorch1.9

今天刚到的r7000p,配置pytorch安装了好久,记录一下安装过程


首先我的环境为cuda11.2,显卡3050ti

首先在官网查看安装命令
RTX3050ti安装pytorch1.9_第1张图片
发现cuda11.x需要从conda-forge安装某些包,此命令是从官网下载的,官网的速度大家也是知道的,所以采用换源安装,但是由于换源没法从conda-forge安装,而且清华源的pytorch1.9速度也特别慢,所以终极解决办法为:

  1. 先换源
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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 --set show_channel_urls yes
  1. 手动从https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/下载pytorch1.9(建议用迅雷
    RTX3050ti安装pytorch1.9_第2张图片
  2. 离线安装pytorch1.9
conda install --use-local pytorch-1.9.0-py3.6_cuda11.1_cudnn8_0.tar.bz2
  1. 最后使用命令安装其他包
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/ -c conda-forge

测试是否安装成功
RTX3050ti安装pytorch1.9_第3张图片

你可能感兴趣的:(anaconda,pytorch,cuda,深度学习,python)