anaconda配置pytorch环境cuda118

以下是在Windows 10环境下使用Anaconda配置PyTorch环境CUDA 11.1的步骤：
1.首先，确保你已经安装了Anaconda，如果没有，请先安装Anaconda。

2.打开Anaconda Prompt，创建一个新的conda环境并激活它：

conda create --name pytorch_env
conda activate pytorch_env

3.安装PyTorch和Torchvision：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

4.测试PyTorch是否安装成功：

python
import torch
print(torch.__version__)
print(torch.cuda.is_available())

如果输出了PyTorch的版本号和True，则说明PyTorch已经成功安装并且可以使用CUDA加速。