这篇博客就是为了记录下我自己配置PyTorch-GPU的过程。
因为我之前换过一次电脑,导致我这里还要再重新配置一下PyTorch。然而我之前配置的时候又没有做好记录,这次配置又要百度各种教程。😂吸取之前的教训,我现在要记录下我自己的配置过程,省的之后再安装还要百度23333333
创建conda虚拟环境
我通过如下命令创建了虚拟环境:
conda create -n pygpu pip python=3.7
在这里我发现我老是无法创建成功,我感觉可能就是源的原因吧。于是我参考 conda源配置 这篇博客,使用如下命令配置中科大源,解决了问题。
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --set show_channel_urls yes