pytorch

pytorch官网:https://pytorch.org/get-started/locally/

目前gpu版pytorch安装不需要自己安装cuda,而且注意windows版仅支持cuda11.3 ,在镜像网站可以找到cuda11.1、11.3、11.5的版本
在这里插入图片描述在这里网络环境好的可以直接使用命令安装,这里是conda安装方式

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

网络环境不好最好还是用镜像网站提供的文件,比如
pytorch:https://download.pytorch.org/whl/torch/
torchversion:https://download.pytorch.org/whl/torchvision/(这个装不装都行)
文件命名就是版本信息,比如:torch-1.11.0+cu115-cp38-cp38-win_amd64.whl,1.11.0是pytorch的版本号,cu115是cuda11.5,cp38就是python3.8

使用whl安装好像只能用pip

pip install 路径+文件名.whl

cuda和驱动

关于cuda,要与显卡驱动版本的匹配
官方的CUDA Toolkit版本和显卡驱动对应的表格。
(注:网站:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html)
在这里插入图片描述
显卡驱动版本在命令行输入nvidia-smi,里面的cuda version不用管
在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐