cuda117 版本支持torch==1.7.1 torchvision==0.2.2在anaconda promt中,cuda11.7的环境中怎么安装上述库
时间: 2024-10-14 15:11:31 浏览: 63
torch_scatter-2.1.2+pt20cu117-cp38-cp38-win_amd64whl.zip
在Anaconda Prompt中,在CUDA 11.7环境下安装`torch==1.7.1` 和 `torchvision==0.2.2` 库,你需要先确保已经安装了Python和对应版本的PyTorch CUDA支持。如果你还未安装,可以按照以下步骤操作:
1. **更新 conda**:
```
conda update conda
```
2. **创建一个新的环境** (建议新建,以防影响其他项目):
```
conda create -n torch_cuda117 python=3.8
```
可以替换`python=3.8`为实际需要的Python版本。
3. **激活新环境**:
```
conda activate torch_cuda117
```
4. **安装 CUDA 和 cuDNN**:
```
conda install cudatoolkit=11.7 cudnn=8.2.0
```
需要下载并提供cuDNN的二进制包,你可以从NVIDIA官网下载适合的cuDNN版本。
5. **安装 PyTorch 和 torchvision**:
```
pip install torch==1.7.1+cu117 torchvision==0.2.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
```
这里使用了PyTorch的特定版本仓库URL,确保能安装到与CUDA兼容的版本。
6. **检查安装**:
```
python -c "import torch; print(torch.__version__, torch.cuda.is_available())"
```
如果一切顺利,应该会显示安装好的`torch`版本,并确认GPU可用。
阅读全文