PyTorch安装

pytorch是非常脆弱的,它的版本和cuda对应好才可以使用,所以cuda并不是越新越好的。pytorch其实是非常好下载安装的,难题就是cuda版本选择的问题。

1 打开Pytorch官网,选择previous-versions,按照要求选择自己的合适版本。注意系统的选择,有英伟达显卡的选择cuda,没有的选择cpu。安装之前看一下自己电脑显卡最高能支持什么版本的cuda!!!!

官网:PyTorch

2 对了,如果有的同学没有换源的先进行换源!!!点这里CSDN

打开Anaconda Promp,进入虚拟环境,输入官网上给出的conda代码。

 3 安装测试

python

import torch

print(torch.cuda.is_available())

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值