本文主要是介绍nvidia a100-pcie-40gb环境安装,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
1.conda create --name torch_li python=3.8
2.conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
环境测试:torch.cuda.is_available()
3.conda remove -n torch_li --all
4.pip install opencv-python-headless
5.pip install matplotlib
6.pip install imageio
删除虚拟环境:conda remove -n torch_li --all
这篇关于nvidia a100-pcie-40gb环境安装的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!