cuda 8.0 安装pytorch遇到的坑

第一步:安装cuda8和cudnn。
这位博主的文章讲的很清楚,这里就不在赘述了:

cuda8.0 cuda 8.0 安装pytorch_qq_42331243的博客-CSDN博客_cuda8.0 pytorch

第二步:安装pytorch。

我这里的情况是安装pytorch时出现问题,离线安装和anaconda安装都不成功,由于水平有限,估计也解决不了,最后我这里选择了miniconda安装的方式。

Miniconda — conda documentation

 miniconda默认源网速很慢,这里需要切换到清华源。

配置文件地址:

 配置内容:

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

配置完成之后就可以安装pytorch了:

Previous PyTorch Versions | PyTorch

 这里给出安装命令:

conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

 第三步:验证安装结果

 

 如果能输出这个结果,应该就表示安装成功了。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值