pytorch和torchvision的安装

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

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

https://pytorch.org/

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

pytorch和torchvision的安装_第1张图片

下面就会出现命令,直接复制过去,前面加 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

 

 

你可能感兴趣的:(pytorch,pytorch)