Ubuntu 1050ti应该安装什么显卡驱动来进行pytorch任务
时间: 2023-07-10 16:05:10 浏览: 167
Ubuntu中安装GPU版本pytorch
5星 · 资源好评率100%
对于Ubuntu 1050ti,建议使用NVIDIA官方提供的显卡驱动程序。您可以按照以下步骤安装:
1. 打开终端并更新系统:
```
sudo apt-get update
sudo apt-get upgrade
```
2. 安装依赖项:
```
sudo apt-get install build-essential dkms linux-headers-$(uname -r)
```
3. 下载适用于您的显卡的NVIDIA驱动程序。您可以从NVIDIA官方网站上下载最新版本的驱动程序。
4. 切换到下载目录并在终端中运行以下命令:
```
sudo chmod +x NVIDIA-Linux-x86_64-xxx.xx.run
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run
```
请注意,在上面的命令中,”xxx.xx”应替换为您下载的驱动程序的版本号。
5. 安装完成后,重新启动计算机。
6. 最后,您可以安装PyTorch并在Python中使用GPU:
```
pip install torch torchvision
```
现在,您的系统已准备好使用PyTorch进行GPU加速任务。
阅读全文