1. 安装Anaconda(环境变量);
2. 创建虚拟环境;
conda create -n UCTransNet-3.7 python=3.7
conda activate UCTransNet-3.7
3. 安装pytorch带GPU
pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/
(带豆瓣的镜像源)
4. 验证是否成功
torch.cuda.is_available()