pytorch安装 古城微笑少年丶 2022-01-25 15:11 282阅读 0赞 安装pytorch之前先查看ubuntu下cuda的版本号,再进行安装 conda install pytorch torchvision cudatoolkit=8.0 -c pytorch 如果安装速度慢的话,去掉-c pytorch。因此“-c pytorch”默认从pytorch官网下载安装包,因此安装的时候将-c pytorch去掉,再修改为清华源,可提高安装速度。 指定python版本 conda install python=2.7 pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch 基于cuda10.1版本的安装 conda install python=2.7 pytorch=1.3.0 torchvision cudatoolkit=10.1 PS: 清华源添加方法: conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge conda config --set show_channel_urls yes 然后系统会自动在/home中增加.condarc文件,包括如下信息: channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge - default show_channel_urls: true 有的简称写到,需要将 - default 那一行删掉,我没有删掉,目前没有什么影响。接着保存.condarc .然后重启终端就能用conda安装新的库了。 替换了源之后,pytorch的安装的源变成了清华源,也就快了很多 ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2h1bGU_size_16_color_FFFFFF_t_70][] [watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2h1bGU_size_16_color_FFFFFF_t_70]: /images/20220125/699cfaa6722749b0867fa2608b522140.png
还没有评论,来说两句吧...