环境准备：虚拟环境搭建

目录

本机环境：

1.conda 相关

2.虚拟环境搭建

3. PyG(PyTorch Geometric) 安装

4.HiVT实验：训练及测试

参考​​​​​​​

---

本机环境：

RTX3060 + cuda11.4 + cudnn8.2.4 + 
GPU环境搭建过程提前下载的文件如下：
 

NVIDIA-Linux-x86_64-470.129.06.run
cudnn-11.4-linux-x64-v8.2.4.15.tgz
cuda_11.4.2_470.57.02_linux.run

1.conda 相关

深度学习环境，建议直接miniconda里搞虚拟环境来玩。

【conda】
创建环境
conda create -n HiVT python=3.8

清除缓存
conda clean -a

安装包
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy

查看环境包信息
要查看当前环境中所有安装了的包可以用

conda list

导入导出环境

如果想要导出当前环境的包信息可以用

conda env export > environment.yaml

将包信息存入yaml文件中.

当需要重新创建一个相同的虚拟环境时可以用

conda env create -f environment.yaml

其实命令很简单对不对, 我把一些常用的在下面给出来, 相信自己多打两次就能记住

activate // 切换到base环境

activate learn // 切换到learn环境