【PyTorch】夏侯南溪的安装教程

1. 安装要求

python < 3.11

2. 使用conda安装PyTorch

Step 1:进入进行安装的conda环境,如:
conda activate conda-general
Step 2:进入PyTorch官网按照说明进行安装,
安装链接:PyTorch-GET-STARTED

Note:
现在的PyTorch在安装时,也会自动帮助我们安装 CUDA Toolkit(pytorch-cuda),所以可以不用再手动地提前安装CUDA了。
关于PyTorch在构建时如何搜索CUDA目录,请参考博文《linux上的,多种安装方式产生的,多个cuda版本,以及执行顺序问题(nvidia-smi,/usr/local/,conda)》

step 3:(如果使用官方命令无法正常安装的话,)
就需要使用命令
conda install pytorch torchvision cudatoolkit=10.2
来进行安装,其实官网给出的命令是这样的
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
不过在实际安装的时候会发现安装不成功,会出现“CondaHTTPError: HTTP 000 CONNECTION FAILED for url ……”的错误,which may be due to the existence of BFW, 所以还需要问一下度娘~
后来就找到了网友夜幕下的光123的博文《Pytorch安装–报错-CondaHTTPError: HTTP 000 CONNECTION FAILED for url》,其中说到如果使用官方的命令“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”进行安装,则不会从清华源进行下载,其解决办法就是去掉“-c pytorch”;

验证

import torch
import torchvision

print(torch.cuda.is_available())

2 Troubleshooting

(1)Conda在“Solving environment:…”这里一直在载入(“光标转圈…”),很长时间都没有响应**

a. 可能是优于conda官方源连接速度太慢,导致conda解析(“Solving”)很慢

.condarc中加入清华源。

b. 使用pip速度会快一些

尝试使用Torch的Pip方式进行安装。

(2)下载速度太慢

  1. 使用conda方式安装可能时间较长,也许需要一个时段左右的时间;
  2. 使用pip方式安装,可以看到下载速度,比较友好;
  3. Try scrambling over the wall.

(3)出现“段错误 (核心已转储)”——“之前由于网速太慢中断过安装”**

需要对之前下载的临时文件进行清理,使用下面的代码

conda clean -a
  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值