anaconda查看pytorch版本cpu gpu
时间: 2024-10-09 07:14:41 浏览: 153
在Anaconda环境中查看PyTorch的CPU和GPU版本,你可以按照以下步骤操作:
1. 打开命令行终端,首先激活你的Anaconda环境。如果你已经激活了环境,可以跳过这一步。
```bash
conda activate your_env_name
```
2. 然后安装`torch`如果还没安装,或者确认已安装的版本:
```bash
pip install torch torchvision
```
3. 安装完成后,你可以通过`pip`或直接导入`torch`模块来查看PyTorch版本:
```python
python
# 或者在命令行里输入
python -c "import torch; print(torch.__version__)"
```
4. 如果你想查看是否支持GPU,以及对应的CUDA版本,可以在Python交互模式下运行下面的代码:
```python
import torch
if torch.cuda.is_available():
device = torch.device("cuda")
print(f"当前设备: {device}")
print(f"CUDA版本: {torch.version.cuda}")
else:
print("当前设备不支持GPU.")
```
5. 这将显示当前可用的硬件、PyTorch的CUDA版本,如果没有CUDA支持,则会显示“当前设备不支持GPU”。
阅读全文
相关推荐

















