nvidia a100-pcie-40gb环境安装

LMengi000

已于 2024-12-03 08:45:56 修改

阅读量582

点赞数 3

文章标签： python

于 2024-03-01 18:16:33 首次发布

本文链接：https://blog.csdn.net/LMengi000/article/details/136403315

版权

安装screen： 1：apt-get install screen 2：apt-get update
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 -i https://pypi.tuna.tsinghua.edu.cn/simple
5.pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
6.pip install imageio -i https://pypi.tuna.tsinghua.edu.cn/simple
7.pip install tensorboardX -i https://pypi.tuna.tsinghua.edu.cn/simple
8.pip install tensorboard -i https://pypi.tuna.tsinghua.edu.cn/simple
9.pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple
10.pip install medpy -i https://pypi.tuna.tsinghua.edu.cn/simple
11.pip install nibabel -i https://pypi.tuna.tsinghua.edu.cn/simple
12.pip install sci