pytorch安装

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

1.安装好Anaconda

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

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

2.安装Pytorch

(1)我的电脑;显卡是NVIDIA RTX3060
看到显卡装有支持粗大11.6的驱动器。
(2)到PyTorch官网上选择你需要安装的环境配置(
在这里插入图片描述
这里因为前面安装的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运行
在这里插入图片描述
如果运行时出现了这个问题:

. 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.测试是否安装成功

在这里插入图片描述
进入python后,运行

import torch
torch.cuda.is_available()

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值