pytorch安装

刚开始学习强化学习,需要安装Pytorch(GPU版本),记录一下自己安装过程。这里我没有单独下载和安装网友说的CUDA,装Pytorch(1.21.1)过程中包含进去了。

  • 1.安装好Anaconda
  • 2.安装Pytorch
  • 3.测试是否安装成功

1.安装好Anaconda

下载最新版本,按部就班的安装即可,没有出现啥问题,这里不再赘述。

(1)为pytorch创建一个环境:conda create -n hhh python=3.9
(2)查看该虚拟环境hhh是否创建成功:
在这里插入图片描述
(3)激活这个环境:
在这里插入图片描述

2.安装Pytorch

(1)我的电脑;显卡是NVIDIA RTX3060
pytorch安装_第1张图片
(2)到PyTorch官网上选择你需要安装的环境配置()
pytorch安装_第2张图片
这里因为前面安装的Anaconda,因此Package选择Conda,并选择对应的CUDA11.6版本。提示11.6需要加入’conda-forge’ channel(个人理解一个下载源)来下载cudatoolkit11.6。参考网络教程中(加入清华源地址),需要添加4个channel的语句:

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
特别是第4个,不能丢啊

查看channel:conda config --show
删除channel:conda config –remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
增加channel:conda config ‐‐add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

(3)把自动生成的命令语句(绿色框内)粘贴到Anaconda Powershell Prompt运行
pytorch安装_第3张图片
如果运行时出现了这个问题:

. HTTP 000 CONNECTION FAILED for url https://conda.anaconda.org/pytorch/win-64/current_repodata.json

An HTTP error ocurred when trying to retrieve this URL.

HTTP errors are often intermittent, and a simple retry will get you on your way.

在知乎上的这篇文章中找到解决方法,大家可以参考:
添加链接描述
很简单:在你安装的Anaconda文件下找到.condarc文件,打开.condarc文件,把https改成http。

(5)然后继续运行

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

然后开始检测要下载的文件,由于太多,我没有来得及保存截图,这里最主要的文件就是pytorch, torchvision ,torchaudio ,cudatoolkit=11.6,总之等了很长时间大概半小时,安装完成了!

3.测试是否安装成功

pytorch安装_第4张图片
进入python后,运行

import torch
torch.cuda.is_available()

如果返回True表示安装成功!

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