记录代码复现过程中pytorch环境搭建的坑

1.首先是python版本的选择:尽量根作者保持一致!高版本的很多不一定兼容低版本,本文代码尝试过3.10、3.8、3.6版本,由于版本不同,安装的pytorch版本也不同,对代码复现很有大影响。其中cuda、python、torch版本对应关系网上很多,比如:记录代码复现过程中pytorch环境搭建的坑_第1张图片

创建虚拟环境: 

conda create-n your_env_name python=x.x

2.最重要的安装torch步骤:

首先踩过的一个坑,由于下载速度过慢,很容易因为网络问题导致下载失败很多时候会选择用清华源进行下载,但是其默认下载的其实是cpu版本,所以需要通过别的途径下载,试过的最有效的办法还是手动下载,步骤可参考:https://blog.csdn.net/qq_46126258/article/details/112708781

主要就是去pytorch官网查看适合你版本的命令,然后通过https://download.pytorch.org/whl/torch_stable.html

选择对应版本的文件下载下来:

本代码需要用到这三个文件:

torch-1.10.2+cpu-cp36-cp36m-linux_x86_64.whl
torchvision-0.11.2+cu113-cp36-cp36m-linux_x86_64.whl
torchaudio-0.10.1+cu113-cp36-cp36m-linux_x86_64.whl

然后就是通过终端pip install 上面的文件了

3.还有一个就是pydensecrf包的安装容易出问题,一般可通过:https://blog.csdn.net/weixin_45209827/article/details/125263406 

解决,主要命令为:

git clone https://github.com/lucasb-eyer/pydensecrf.git
pip install -e pydensecrf

记得如果还是装不上换个网络试试,我是连的手机热点。

其他的安装包就按正常流程安装就好了,觉得网速慢就用镜像源,例如:

# 使用pip指定从清华镜像源中安装pandas库
sudo pip install pandas==1.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

也可以不加sudo

你可能感兴趣的:(语义分割代码复现,pytorch,python,人工智能)