cuda10.1安装torch1.7.1和torchvision0.8.2
时间: 2023-09-05 10:13:05 浏览: 104
torch1.7+torchvision-0.8.rar
您可以按照以下步骤来安装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在内的最新版本。希望这可以解决您的问题!
阅读全文