Anaconda3安装pytorch-gpu避坑(Windows版)

一、Anaconda3安装注意事项(避坑一)

在安装Anaconda3时,用户选择时要注意选择Just Me,否则后续安装步骤将无法勾选添加环境变量的选项,导致安装完成后需要手动配置环境变量。

 手动配置环境变量则需在Path中添加如下(假设安装在了D:\anaconda目录下):

D:\anaconda
D:\anaconda\Scripts\
D:\anaconda\Library\bin
D:\anaconda\Library\mingw-w64\bin

二、cuda及cuDNN安装(避坑二)

cuda安装主要是安装与自己电脑显卡相匹配的版本即可,但是cuda安装过程中一定要注意以下几点:

1、一定要选择自定义安装!!!

 

2、Visual Studio Integration一定不要勾选,否则将安装失败!!!

 3、如果本机的驱动版本(即上图中的当前版本)小于或等于cuda对应的版本(即上图中的新版本),则选择该组件,否则千万不要选!!!

4、最后附上大佬的详细安装教程。

cuda安装 (windows版)

三、Anaconda Prompt安装pytorch(避坑三)

1、在安装pytorch之前一般需要创建虚拟环境,否则直接装在了base环境下,创建步骤如下:

# 创建虚拟环境
conda create -n 虚拟环境名称 python=x.x
# 激活虚拟环境
conda activate 虚拟环境名称

2、创建好虚拟环境并激活后,一般来说直接执行pytorch官网安装命令就行,如下:

但由于网络等原因下载速度较慢(我第一次下载很快,第二次很慢), 所以很多人会选择使用清华镜像,镜像安装命令如下(在base环境或激活环境中执行均可):

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --set show_channel_urls yes

此时就是第三个坑了,一定要注意是http而不是https,否则安装到最后会报网络错误!!!

四、Anaconda Prompt安装pytorch,部分包下载失败的处理(避坑四)

很多人在安装过程中遇到了卡顿,导致个别包没有下载完,后续重新下载又遇到安装失败的错误,这是由于没有清理之前下载失败的包,导致了冲突。

建议先删除虚拟环境下所有的包,再重新下载安装,命令如下:

conda clean --all

注意:此命令会删除所有下载的包,一定要在创建的虚拟环境中执行,否则其他环境也得重新下载。

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

穿我衣服睡我床

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值