pytorch和torchvision的安装

在pycharm中使用清华或中科院的源,找不到匹配的版本,因此使用下面的一行命令即可安装。

在安装之前,需要确定一下自己的一些信息,比如CUDA版本,或者是通过Pip还是Conda安装,是python还是C++/java,查看自己的cuda版本使用命令:nvcc -V,注意大小写和空格,然后打开官网:

https://pytorch.org/

找到:Quick Start Locally,选择自己的版本:

下面就会出现命令,直接复制过去,前面加 sudo即可,这样可以最大限度的保证你不会安装错误的版本。当然没有安装CUDA的可以先更新自己的GPU驱动然后按照CUDA,不过相信大部分同学拿到GPU后肯定会拼命的先装上CUDA。没有GPU的直接选择None即可。

网速慢的,可以打开推荐命令后面的连接,找到对应版本先下载下来,然后再到下载好的文件夹内pip install xxx.whl。

如果GPU不可用,对照这个链接找原因:

https://blog.csdn.net/yinxian9019/article/details/106763892

更换CUDNN版本

https://blog.csdn.net/qq_40755643/article/details/95975201

 

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值