pytorch-gpu安装

要安装pytorch-gpu，您可以按照以下步骤进行操作：

1. 打开Anaconda Prompt。
2. 创建一个新的Python虚拟环境。您可以使用以下命令：conda create -n pytorch_gpu python=3.7
这将创建一个名为"pytorch_gpu"的环境。
3. 等待环境创建完成后，激活pytorch_gpu环境。您可以使用以下命令：conda activate pytorch_gpu
如果前面括号的内容发生改变，表示您已成功进入pytorch_gpu环境。
4. 安装pytorch和torchvision。根据您的CUDA版本选择对应的命令：

- 如果您选择的是CUDA 10.2版本，在pytorch_gpu环境中使用以下命令：
pip3 install -i https://pypi.tuna.tsinghua.edu.＊＊＊.pytorch.org/whl/torch_stable.html

- 如果您选择的是CUDA 10.1版本，在pytorch_gpu环境中使用以下命令：
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

请注意，在安装时要删除命令中-c pytorch后面的内容，以从国内源进行下载以提高下载速度。

通过按照上述步骤操作，您将能够成功安装pytorch-gpu。