怎么下载不是专门为CUDA 10.1版本优化torch1.7.1
时间: 2023-05-19 10:04:21 浏览: 47
您可以尝试在官方网站上下载适用于CUDA 10.1版本的torch1.7.1,或者使用conda或pip安装torch1.7.1并指定CUDA版本。如果您仍然遇到问题,您可以在相关的论坛或社区中寻求帮助。
相关问题
linux用conda重新安装cuda10.1对应的torch1.7.1GUP版本的命令
以下是在 Linux 上使用 Conda 重新安装 CUDA 10.1 对应的 Torch 1.7.1 GPU 版本的命令:
1. 创建一个新的 conda 环境:
```
conda create --name torch-env
```
2. 激活 conda 环境:
```
conda activate torch-env
```
3. 安装 CUDA 10.1:
```
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
```
4. 安装 Torch 1.7.1 GPU 版本:
```
pip3 install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
这些命令将会在您的系统上安装 CUDA 10.1 和 Torch 1.7.1 GPU 版本。请确保您的系统已经正确安装了 NVIDIA 显卡驱动和 CUDA。
cuda10.1安装torch1.7.1和torchvision0.8.2
您可以按照以下步骤来安装cuda10.1并安装torch1.7.1和torchvision0.8.2:
1. 首先,使用以下命令下载cuda10.1 runfile:
```
wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
```
2. 接下来,运行以下命令以使runfile可执行并运行它:
```
chmod +x cuda_10.1.243_418.87.00_linux.run
sudo ./cuda_10.1.243_418.87.00_linux.run
```
3. 稍等片刻,按照提示进行安装,包括选择安装路径和安装选项。
4. CUDA安装完成后,您可以使用以下命令安装Torch和Torchvision:
```
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
这将安装包括Torch和Torchvision在内的最新版本。希望这可以解决您的问题!