设置 StyleMapGAN 所需要的环境

标签: 环境搭建 发布于:2021-12-13 16:28:52 编辑于:2021-12-13 16:54:44 浏览量:1540

Linux 下的环境设置

  1. 下载并运行:cuda_11.1.0_455.23.05_linux.run
  2. 配置环境变量:
    export CUDA_HOME=/usr/local/cuda
    export PATH="${CUDA_HOME}/bin:$PATH"
    export LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64:$LD_LIBRARY_PATH"
    
  3. 安装 PyTorch:conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

注意:

  1. 由于是 30 系显卡,CUDA 版本最低为 11.1;
  2. 其余依赖随便装即可。
  3. 如果运行时程序卡住,去 ~/.cache/torch_extensions/ 目录下删掉 lock 文件。

Windows 下的环境设置

  1. 下载并安装 CUDA 10.1,并设置相应的环境变量,这里不再赘述:
    C:\Users\song>nvcc --version
    nvcc: NVIDIA (R) Cuda compiler driver
    Copyright (c) 2005-2019 NVIDIA Corporation
    Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
    Cuda compilation tools, release 10.1, V10.1.243
    
  2. 打开 Visual Studio Installer,确保以下组件已安装:
    1. MSVC v142 - VS 2019 C++/x86 build tools (Latest);
    2. Windows 10 SDK;
    3. Windows Universal CRT SDK。
  3. 设置环境变量:
    1. PATH 里追加:D:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx64\x64
    2. LIB 里追加:D:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\lib\x64;C:\Program Files (x86)\Windows Kits\10\Lib\10.0.10240.0\ucrt\x64
    3. INCLUDE 里追加:D:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\include;C:\Program Files (x86)\Windows Kits\10\Include\10.0.10240.0\ucrt

未经允许,禁止转载,本文源站链接:https://iamazing.cn/