From a527a7765607c8115fa75b877de0599998eb6eb8 Mon Sep 17 00:00:00 2001
From: Turgut <turqut.a.314@gmail.com>
Date: Thu, 11 Apr 2024 12:21:15 +0400
Subject: [PATCH] update torch versions

---
 NEWS.md               |  4 ++++
 R/install_fastai.R    | 18 +++++++++---------
 man/install_fastai.Rd |  2 +-
 3 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/NEWS.md b/NEWS.md
index ced4dc62e..5e09ee692 100644
--- a/NEWS.md
+++ b/NEWS.md
@@ -1,4 +1,8 @@
 
+## 2.2.2
+
+- Updated installation function ```install_fastai```
+
 ## 2.2.1
 
 - Fixed breaking changes during model training
diff --git a/R/install_fastai.R b/R/install_fastai.R
index 57d8ba356..ab62b2f30 100644
--- a/R/install_fastai.R
+++ b/R/install_fastai.R
@@ -10,7 +10,7 @@
 #' @param TPU official way to install Pytorch-XLA 1.13
 #' @return None
 #' @export
-install_fastai <- function(version, gpu = FALSE, cuda_version = '11.6', overwrite = FALSE,
+install_fastai <- function(version, gpu = FALSE, cuda_version = '11.8', overwrite = FALSE,
                            extra_pkgs = c('timm','fastinference[interp]'),
                            TPU = FALSE) {
   # extensions
@@ -62,15 +62,15 @@ install_fastai <- function(version, gpu = FALSE, cuda_version = '11.6', overwrit
   }
 
   # linux
-  cuda_linux = c('torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116',
+  cuda_linux = c('torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118',
                  'torch torchvision torchaudio')
   linux_cpu = c('torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu')
 
-  xla = "cloud-tpu-client==0.10 torch==1.13.0 https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-1.13-cp38-cp38-linux_x86_64.whl tensorboard-plugin-profile"
+  xla = "torch~=2.2.0 torch_xla[tpu]~=2.2.0 -f https://storage.googleapis.com/libtpu-releases/index.html"
 
   # windows
-  cuda_windows = c('torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116',
-                   'torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117')
+  cuda_windows = c('torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118',
+                   'torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121')
   cpu_windows = c('torch torchvision torchaudio')
 
   if('torch' %in% required_py_pkgs ) {
@@ -107,10 +107,10 @@ install_fastai <- function(version, gpu = FALSE, cuda_version = '11.6', overwrit
     if(py_av) {
 
       if (os() %in% 'linux' & !length(required_py_pkgs) == 0 & !TPU) {
-        if (os() %in% 'linux' & gpu & cuda_version %in% '11.6' & torch_r & !length(required_py_pkgs) == 0) {
+        if (os() %in% 'linux' & gpu & cuda_version %in% '11.8' & torch_r & !length(required_py_pkgs) == 0) {
            fastai_py_install( c(required_py_pkgs, cuda_linux[1]) )
 
-        } else if (os() %in% 'linux' & gpu & cuda_version %in% '11.7' & torch_r & !length(required_py_pkgs) == 0) {
+        } else if (os() %in% 'linux' & gpu & cuda_version %in% '12.1' & torch_r & !length(required_py_pkgs) == 0) {
           fastai_py_install(c(required_py_pkgs, cuda_linux[2]) )
 
         } else if(!gpu & torch_r & !length(required_py_pkgs) == 0) {
@@ -134,10 +134,10 @@ install_fastai <- function(version, gpu = FALSE, cuda_version = '11.6', overwrit
       }
 
       if (os() %in% 'windows' & !length(required_py_pkgs) == 0 & torch_r & !length(required_py_pkgs) == 0) {
-        if (os() %in% 'windows' & gpu & cuda_version %in% '11.6' & torch_r & !length(required_py_pkgs) == 0) {
+        if (os() %in% 'windows' & gpu & cuda_version %in% '11.8' & torch_r & !length(required_py_pkgs) == 0) {
           fastai_py_install( c(required_py_pkgs, cuda_windows[1]) )
 
-        } else if (os() %in% 'windows' & gpu & cuda_version %in% '11.7' & torch_r & !length(required_py_pkgs) == 0) {
+        } else if (os() %in% 'windows' & gpu & cuda_version %in% '12.1' & torch_r & !length(required_py_pkgs) == 0) {
          fastai_py_install( c(required_py_pkgs, cuda_windows[2]) )
 
         } else if(!gpu & torch_r & !length(required_py_pkgs) == 0) {
diff --git a/man/install_fastai.Rd b/man/install_fastai.Rd
index 7b8e6ea6c..c5240018f 100644
--- a/man/install_fastai.Rd
+++ b/man/install_fastai.Rd
@@ -7,7 +7,7 @@
 install_fastai(
   version,
   gpu = FALSE,
-  cuda_version = "11.6",
+  cuda_version = "11.8",
   overwrite = FALSE,
   extra_pkgs = c("timm", "fastinference[interp]"),
   TPU = FALSE