当前位置:   article > 正文

服务器配置环境记录_conda install cudatoolkit=10.0

conda install cudatoolkit=10.0

l参考 https://blog.csdn.net/qq_42845932/article/details/123964867?spm=1001.2014.3001.5502

conda

下载地址:

https://docs.conda.io/en/latest/miniconda.html

修改源:

https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

参考https://blog.csdn.net/innovationy/article/details/121510666

  1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  2. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  3. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  4. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  5. conda config --set show_channel_urls yes
  6. 完成后输入命令vim ~/.condarc查看配置文件
  7. 下图是配置文件内容,按键盘把按键光标移到defaults处,将最后一行defaults删去,否则还会在官方库中寻找。
  8. conda update --all
  9. conda update -n base conda

安装cuda

  1. # conda install cudatoolkit=10.0.0
  2. conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

安装cudnn

conda install cudnn=7.6

安装pytorch(https://pytorch.org/get-started/previous-versions/

  1. pip install torch==1.2.0 torchvision==0.4.0
  2. 安装完conda以及创建好环境后或者直接安装:
  3. conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
  4. 或者用pip3安装,速度更快一些。

验证

  1. import torch
  2. flag = torch.cuda.is_available()
  3. if flag:
  4.     print("CUDA可使用")
  5. else:
  6.     print("CUDA不可用")

挂载

  1. ln -s /data/xupin/DFUC2022 /home/xupin/dfuc2022
  2. ln -s /data/xupin/save_weights /home/xupin/dfuc2022

声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号