CUDA10.1 下好pytorch安装包后如何配置pytorch
时间: 2023-04-09 10:01:20 浏览: 247
CUDA10.1向下兼容10.0,完美运行官方TensorFlow、Pytorch_pytorchcuda10.0,cuda10.0可以升级
5星 · 资源好评率100%
您可以按照以下步骤配置PyTorch:
1. 确保您已经安装了CUDA 10.1和对应的cuDNN版本。
2. 打开Anaconda Prompt或终端,并创建一个新的虚拟环境。
3. 在虚拟环境中安装PyTorch,可以使用以下命令:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
4. 安装完成后,您可以在Python中导入PyTorch并开始使用它。
希望这可以帮助您配置PyTorch。
阅读全文