conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
时间: 2023-09-16 12:09:46 浏览: 121
树莓派 pytorch1.8.1 + torchvision0.9 对应的arrch64位文件
这是一个用于安装 PyTorch 1.8.0、torchvision 0.9.0 和 torchaudio 0.8.0 的命令,同时指定了使用的 CUDA 版本为 11.1。它从两个不同的 channel(pytorch 和 conda-forge)中获取软件包。你可以在 Anaconda 或 Miniconda 中的命令行中运行这个命令来安装这些软件包。
阅读全文