本文主要是介绍【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 torchvisionprint(torch.cuda.is_available())
2 Troubleshooting
(1)Conda在“Solving environment:…”这里一直在载入(“光标转圈…”),很长时间都没有响应**
a. 可能是优于conda官方源连接速度太慢,导致conda解析(“Solving”)很慢
在.condarc
中加入清华源。
b. 使用pip速度会快一些
尝试使用Torch的Pip方式进行安装。
(2)下载速度太慢
- 使用
conda
方式安装可能时间较长,也许需要一个时段左右的时间; - 使用
pip
方式安装,可以看到下载速度,比较友好; - Try scrambling over the wall.
(3)出现“段错误 (核心已转储)”——“之前由于网速太慢中断过安装”**
需要对之前下载的临时文件进行清理,使用下面的代码
conda clean -a
这篇关于【PyTorch】夏侯南溪的安装教程的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!