- 创建conda环境。conda create -n name python==3.6
- 配置pytorch环境。过老的pytorch1.0不好配。采用1.8也可以完成。
-
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
- 检查版本:
>>> import torch >>> torch.version.cuda '11.1' >>> torch.__version__ '1.8.0' >>> torch.cuda.is_available() True >>> torch.cuda.device_count() 1
- 安装其他必备包:
pip install --upgrade pip pip install tqdm scipy easydict scikit-image pyyaml==5.1 shapely tensorboardX numba
- 更改部分版本导致的代码错误:
PointRCNN代码逐步复现
于 2024-01-28 20:30:57 首次发布