nvidia a100-pcie-40gb环境安装

原创

已于 2024-12-03 08:45:56 修改 · 697 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

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

安装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 scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple