gpu版pytorch安装
·
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不用管

更多推荐
所有评论(0)