CUDA11.4安装gpu版pytorch
时间: 2023-10-25 20:13:47 浏览: 89
以下是在Windows系统上安装GPU版PyTorch的步骤:
1. 安装CUDA11.4
在CUDA官网下载CUDA11.4的安装程序,并按照提示完成安装。安装完成后,将CUDA的bin和lib路径添加到环境变量中。
2. 安装cuDNN
在NVIDIA官网下载cuDNN的安装程序,并按照提示完成安装。安装完成后,将cuDNN的bin和lib路径添加到环境变量中。
3. 安装Anaconda
在Anaconda官网下载Anaconda的安装程序,并按照提示完成安装。
4. 创建虚拟环境
打开Anaconda Prompt,执行以下命令创建一个名为pytorch的虚拟环境:
```
conda create -n pytorch python=3.8
```
5. 激活虚拟环境
执行以下命令激活pytorch虚拟环境:
```
conda activate pytorch
```
6. 安装PyTorch
执行以下命令安装PyTorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
7. 验证安装
执行以下命令,验证PyTorch是否安装成功:
```
python
import torch
print(torch.cuda.is_available())
```
如果输出结果为True,则说明PyTorch安装成功,并且GPU也已经被正确识别。
相关问题
cuda11.4对应python和pytorch版本
CUDA 11.4 目前支持的 Python 版本是 3.6、3.7、3.8 和 3.9,PyTorch 的官方文档中推荐使用 CUDA 11.1 或 11.2,但也支持使用 CUDA 11.4。因此,可以使用 Python 3.6-3.9 和 PyTorch 1.9 来配合 CUDA 11.4 使用。
cuda11.4对应的所有pytorch版本有哪些
您好,根据我的了解,CUDA 11.4 对应的 PyTorch 版本有以下几个:
- PyTorch 1.9.
- PyTorch 1.8.1
- PyTorch 1.7.1
- PyTorch 1.6.
希望能对您有所帮助。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)