前言

本来想跑2020年6月份github上新出的YOLOv5,奈何电脑配置跟不上(我想大声哭泣)。目前YOLOv5的配置要求torch>=1.6(即CUDA为9.2、10.1、10.2以上),然而我安装的CUDA10.0…,把torch版本更新以后就和CUDA对应不上了,所以我还得重新更新电脑驱动,重装CUDA? 我选择放弃。另外一个原因是图2,通过观察内存发现,数据加载过程中内存会被耗尽,机械内存不够,我还没加数据集呢…

UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10000). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at  /pytorch/c10/cuda/CUDAFunctions.cpp:109.)
  return torch._C._cuda_getDeviceCount() > 0

在这里插入图片描述RuntimeError: DataLoader worker (pid 3547) is killed by signal: Killed.
在这里插入图片描述
所以我选择删除重装pytorch,新建python虚拟环境,重新跑一个YOLOv4的代码,安装选择对应代码所需要的环境。
我是利用Conda创建的虚拟环境,pip删除方式用不了。

一、删除python虚拟环境

1.查看当前存在那些虚拟环境

输入以下命令,可以看出当前存在两个虚拟环境,一个是我安装Anaconda时自带的环境,另外一个是自己conda 创建的虚拟环境。

$ conda env list

在这里插入图片描述

2.删除虚拟环境

输入以下命令,将我创建的虚拟环境pytorch_env删除。

$ conda remove -n pytorch_env(虚拟环境名称)  --all
3.确认

最后,输入 conda env list,再次查看当前存在那些虚拟环境。至此,所创虚拟环境已被删除。对应安装的python3.7也没有了。
在这里插入图片描述
在这里插入图片描述

二、删除及安装Anaconda

由于Anaconda的版本号与我即将要新安装的python版本不匹配,所以还得卸载原来python3.7对应的Anaconda3-5.3.0,重新安装适合我将用python对应的版本号。

1.删除Anaconda
1). 删除整个anaconda目录:

由于Anaconda的安装文件都包含在一个目录中,所以直接将该目录删除即可。到包含整个anaconda目录的文件夹下,删除整个Anaconda目录:

rm -rf anaconda文件夹名

2). 建议清理下.bashrc中的Anaconda路径:

a.到根目录下,打开终端并输入:
sudo gedit ~/.bashrc

b.在.bashrc文件末尾用#号注释掉之前添加的路径(或直接删除):
#export PATH=/home/lq/anaconda3/bin:$PATH
保存并关闭文件

c.使其立即生效,在终端执行:
source ~/.bashrc

d.关闭终端,然后再重启一个新的终端,这一步很重要,不然在原终端上还是绑定有anaconda3.

2.安装Anaconda

见我之前发的博客。

三、重新安装虚拟环境

1.虚拟环境的创建命令如下
$ conda create -n your_env_name(虚拟环境名称) python==xx(想要创建的虚拟环境的python版本号)

可以根据自己需要修改
例如要创建基于python3.6版本的python

输入y按下enter确认开始下载安装。

终端提示上述信息说明虚拟环境创建成功。

2.虚拟环境的激活与切换

虚拟环境创建完成之后,终端输入以下命令即可激活虚拟环境

conda activate your_env_name(虚拟环境名称)

三、安装Pytorch

参考以下博主
https://blog.csdn.net/m0_37758063/article/details/110734347?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522161503794216780357240994%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=161503794216780357240994&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allfirst_rank_v2~rank_v29-2-110734347.pc_search_result_hbase_insert&utm_term=pythorch%E5%AE%89%E8%A3%85%E6%95%99%E7%A8%8BUBUNTU


虚拟环境相关命令:

  1. 查看当前存在哪些虚拟环境:conda env list 或 conda info -e;
  2. 查看安装了哪些包: conda list;
  3. 退出虚拟环境: conda deactivate;
  4. 卸载虚拟环境 :conda remove -n your_env_name(虚拟环境名称) --all;
  5. 创建虚拟环境:conda create -n your_env_name(虚拟环境名称);
  6. 激活虚拟环境:conda activate your_env_name(虚拟环境名称);
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐